groloch commited on
Commit
7a1a375
·
verified ·
1 Parent(s): 326c445

Upload folder using huggingface_hub

Browse files
Files changed (5) hide show
  1. dit.pth +3 -0
  2. dit_config.json +12 -0
  3. encoder_config.json +3 -0
  4. vae.pth +3 -0
  5. vae_config.json +21 -0
dit.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:992de40836906ad7c4565d2052f149c229cc88d8d6411abc085f51b6b4ffc9c7
3
+ size 406159595
dit_config.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "mmdit",
3
+ "latent_dim": 16,
4
+ "latent_size": 16,
5
+ "hidden_dim": 512,
6
+ "n_layers": 16,
7
+ "n_heads": 8,
8
+ "text_embed_dim": 768,
9
+ "patch_size": 128,
10
+ "sine_encoding_frequency": 10000,
11
+ "num_classes": 1081
12
+ }
encoder_config.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ {
2
+ "encoder_path": "google/embeddinggemma-300m"
3
+ }
vae.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:94231d899b54ecad421904f3de3115b3d1c63a8496c775216eb4b17a89572055
3
+ size 53156779
vae_config.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "in_channels": 3,
3
+ "image_size": 128,
4
+ "latent_dim": 16,
5
+ "depths": [
6
+ 3,
7
+ 5,
8
+ 7,
9
+ 3
10
+ ],
11
+ "dims": [
12
+ 48,
13
+ 96,
14
+ 192,
15
+ 384
16
+ ],
17
+ "ln_eps": 1e-06,
18
+ "iaf": false,
19
+ "iaf_n_blocks": null,
20
+ "iaf_timesteps": null
21
+ }