AbstractPhil commited on
Commit
cd94a89
·
verified ·
1 Parent(s): 7b370f3

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +6 -18
config.json CHANGED
@@ -1,34 +1,22 @@
1
  {
2
  "architecture": "beatrix_flow_flux_style",
3
  "image_size": 256,
4
- "latent_channels": 4,
5
  "latent_size": 32,
6
  "text_raw_dim": 768,
7
- "hidden_size": 2048,
8
- "tower_dim": 512,
9
- "tower_depth": 1,
10
  "num_heads": 8,
11
  "geometric_types": [
12
  "cantor",
13
- "beatrix",
14
- "helix",
15
- "simplex",
16
- "cantor",
17
- "beatrix",
18
- "helix",
19
- "simplex"
20
  ],
21
  "conv_types": [
22
  "wide_resnet",
23
- "frequency",
24
- "bottleneck",
25
- "squeeze_excite",
26
- "wide_resnet",
27
- "frequency",
28
- "bottleneck",
29
  "squeeze_excite"
30
  ],
31
- "manifold_dim": 1024,
32
  "num_flow_steps": 50,
33
  "num_latent_tokens": 1024
34
  }
 
1
  {
2
  "architecture": "beatrix_flow_flux_style",
3
  "image_size": 256,
4
+ "latent_channels": 16,
5
  "latent_size": 32,
6
  "text_raw_dim": 768,
7
+ "hidden_size": 256,
8
+ "tower_dim": 256,
9
+ "tower_depth": 2,
10
  "num_heads": 8,
11
  "geometric_types": [
12
  "cantor",
13
+ "beatrix"
 
 
 
 
 
 
14
  ],
15
  "conv_types": [
16
  "wide_resnet",
 
 
 
 
 
 
17
  "squeeze_excite"
18
  ],
19
+ "manifold_dim": 512,
20
  "num_flow_steps": 50,
21
  "num_latent_tokens": 1024
22
  }