Hzfinfdu commited on
Commit
578ae20
1 Parent(s): 97f3ba1

Rename folders

Browse files
Llama3_1Base-L4A-8x/checkpoints/final.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:6a469fcba27938a8c81f179b31eb83889d0059edc626f7d86abb3db718c350a5
3
- size 536945024
 
 
 
 
Llama3_1Base-L4A-8x/hyperparams.json DELETED
@@ -1,35 +0,0 @@
1
- {
2
- "device": "cuda:0",
3
- "seed": 42,
4
- "dtype": "torch.bfloat16",
5
- "hook_point_in": "blocks.4.hook_attn_out",
6
- "hook_point_out": "blocks.4.hook_attn_out",
7
- "use_decoder_bias": true,
8
- "apply_decoder_bias_to_pre_encoder": false,
9
- "expansion_factor": 8,
10
- "d_model": 4096,
11
- "d_sae": 32768,
12
- "bias_init_method": "all_zero",
13
- "act_fn": "jumprelu",
14
- "jump_relu_threshold": 0.0546875,
15
- "norm_activation": "dataset-wise",
16
- "dataset_average_activation_norm": {
17
- "in": 1.3984375,
18
- "out": 1.3984375
19
- },
20
- "decoder_exactly_fixed_norm": false,
21
- "sparsity_include_decoder_norm": true,
22
- "use_glu_encoder": false,
23
- "init_decoder_norm": 0.5,
24
- "init_encoder_norm": null,
25
- "init_encoder_with_decoder_transpose": true,
26
- "lp": 1,
27
- "l1_coefficient": 8e-05,
28
- "l1_coefficient_warmup_steps": 39062,
29
- "top_k": 50,
30
- "k_warmup_steps": 39062,
31
- "use_batch_norm_mse": true,
32
- "use_ghost_grads": false,
33
- "tp_size": 1,
34
- "ddp_size": 1
35
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
Llama3_1Base-L4A-8x/lm_config.json DELETED
@@ -1,10 +0,0 @@
1
- {
2
- "device": "cuda",
3
- "seed": 42,
4
- "dtype": "torch.bfloat16",
5
- "model_name": "meta-llama/Meta-Llama-3.1-8B",
6
- "use_flash_attn": false,
7
- "cache_dir": null,
8
- "d_model": 4096,
9
- "local_files_only": false
10
- }