pellement99 commited on
Commit
b61a1e9
·
verified ·
1 Parent(s): 8f221b3

Upload folder using huggingface_hub

Browse files
patch_topk_expansion_factor4_k64_multi_topkFalse_auxk_alpha0.015_full_pipeline/config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"sae": {"expansion_factor": 4, "normalize_decoder": true, "num_latents": 0, "k": 64, "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.015, "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_factor4_k64_multi_topkFalse_auxk_alpha0.015_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_factor4_k64_multi_topkFalse_auxk_alpha0.015_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_factor4_k64_multi_topkFalse_auxk_alpha0.015_full_pipeline/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eda54d39b11a2e07baa2579817564489de3387d9701622983846b98e0e9322c2
3
+ size 104913106
patch_topk_expansion_factor4_k64_multi_topkFalse_auxk_alpha0.015_full_pipeline/state.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b58a715ae652ce9b4f2f8a5d68bedab18a0f7b0f521279c2b34c084febb01c32
3
+ size 42130
patch_topk_expansion_factor4_k64_multi_topkFalse_auxk_alpha0.015_full_pipeline/unet.up_blocks.1.attentions.1/cfg.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"expansion_factor": 4, "normalize_decoder": true, "num_latents": 0, "k": 64, "batch_topk": false, "sample_topk": false, "input_unit_norm": false, "multi_topk": false, "d_in": 1280}
patch_topk_expansion_factor4_k64_multi_topkFalse_auxk_alpha0.015_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:899512c460ede9ef06f739eb6aa1b7b024998fc511cfd7dcb086cf51f7001834
3
+ size 52454728