pellement99 commited on
Commit
46d52e6
·
verified ·
1 Parent(s): 80f1b16

Upload folder using huggingface_hub

Browse files
patch_topk_expansion_factor8_k96_multi_topkFalse_auxk_alpha0.002_full_pipeline/config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"sae": {"expansion_factor": 8, "normalize_decoder": true, "num_latents": 0, "k": 96, "batch_topk": false, "sample_topk": false, "input_unit_norm": false, "multi_topk": false}, "dataset_path": ["/w/nobackup/385/scratch-space/expires-2025-Nov-21/pellement/SAeUron/activations/laion-coco-aesthetic/full_pipeline/output"], "effective_batch_size": 4096, "num_workers": 4, "persistent_workers": true, "prefetch_factor": 2, "grad_acc_steps": 1, "micro_acc_steps": 1, "lr": 0.0004, "lr_scheduler": "linear", "lr_warmup_steps": 0, "auxk_alpha": 0.002, "dead_feature_threshold": 10000000, "feature_sampling_window": 100, "hookpoints": ["unet.up_blocks.1.attentions.1"], "distribute_modules": false, "save_every": 5000, "log_to_wandb": true, "run_name": "patch_topk_expansion_factor8_k96_multi_topkFalse_auxk_alpha0.002_full_pipeline", "wandb_log_frequency": 500, "wandb_project": "sae_stable-diffusion-v1-4", "mixed_precision": "no", "max_examples": null, "seed": 42, "device": "cuda", "num_epochs": 15}
patch_topk_expansion_factor8_k96_multi_topkFalse_auxk_alpha0.002_full_pipeline/lr_scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:04d6e0c3f1da6c77aa218446212ab173e331eedfeae822c363c80b139a2cd92c
3
+ size 1012
patch_topk_expansion_factor8_k96_multi_topkFalse_auxk_alpha0.002_full_pipeline/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fbe8bf659d06df45ab84e0536a41e6c21c31cc5385f54e4156274b75d6837e4c
3
+ size 209811666
patch_topk_expansion_factor8_k96_multi_topkFalse_auxk_alpha0.002_full_pipeline/state.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d776f9ac9b6241dcf75df8dbf9a5ffe121bfd54734cd1468fe1fbfdca0fb25e6
3
+ size 83090
patch_topk_expansion_factor8_k96_multi_topkFalse_auxk_alpha0.002_full_pipeline/unet.up_blocks.1.attentions.1/cfg.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"expansion_factor": 8, "normalize_decoder": true, "num_latents": 0, "k": 96, "batch_topk": false, "sample_topk": false, "input_unit_norm": false, "multi_topk": false, "d_in": 1280}
patch_topk_expansion_factor8_k96_multi_topkFalse_auxk_alpha0.002_full_pipeline/unet.up_blocks.1.attentions.1/sae.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7ec9e27e2390f850630282dae6d6a444e5aa1c2cbadadf30ecf58fecd8b7ccc7
3
+ size 104904008