Upload config.json with huggingface_hub

config.json

@@ -1,10 +1,10 @@
 {
+  "architecture": "beatrix_flow_flux_style",
   "image_size": 256,
-  "num_classes": 10,
   "latent_channels": 4,
   "latent_size": 32,
-  "text_raw_dim": 2048,
-  "bottleneck_dim": 256,
+  "text_raw_dim": 768,
+  "hidden_size": 256,
   "tower_dim": 256,
   "tower_depth": 2,
   "num_heads": 8,
@@ -16,8 +16,7 @@
     "wide_resnet",
     "squeeze_excite"
   ],
-  "conv_spatial_size": 8,
   "manifold_dim": 512,
-  "fingerprint_dim": 64,
-  "num_flow_steps": 50
+  "num_flow_steps": 50,
+  "num_latent_tokens": 1024
 }