Lexa commited on
Commit
dd0da24
·
1 Parent(s): 4a648aa

Upload 2000-step checkpoint

Browse files
Files changed (4) hide show
  1. .gitignore +2 -0
  2. config.json +22 -0
  3. model.safetensors +3 -0
  4. training_args.bin +3 -0
.gitignore ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ logs/
2
+ *.pt
config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "AdaLN_Timestep_Embed_Dim": 256,
3
+ "architectures": [
4
+ "LexaLCM"
5
+ ],
6
+ "cfg_scale": 0.0,
7
+ "d_ff": 6144,
8
+ "d_latent": 1024,
9
+ "d_model": 1536,
10
+ "denoiser_iterations_inference": 40,
11
+ "denoiser_iterations_pretrain": 80,
12
+ "dropout_context": 0.1,
13
+ "dropout_denoiser": 0.05,
14
+ "dropout_latent": 0.1,
15
+ "input_dim": 1024,
16
+ "model_type": "lexa_lcm_pre1",
17
+ "n_heads": 16,
18
+ "num_context_layers": 3,
19
+ "num_denoiser_layers": 4,
20
+ "torch_dtype": "float32",
21
+ "transformers_version": "4.51.3"
22
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c8a3bee0fe8a91731018fdf39a4de156a34bf7162b09f7b1e31d774c267d7f20
3
+ size 1579618660
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dc437a801f32a9d492a6b2db67070beb5bc616174e35eba8c29c9bc6eb7f11fe
3
+ size 5649