transformer/config.json

{
  "_class_name": "GeoCoreTransformer2DModel",
  "_diffusers_version": "0.32.0",
  "in_channels": 128,
  "context_in_dim": 4096,
  "hidden_size": 4096,
  "num_heads": 32,
  "depth": 8,
  "depth_repa": 8,
  "depth_single_blocks": 24,
  "axes_dim": [
    32,
    48,
    48
  ],
  "theta": 2000,
  "mlp_ratio": 3.0,
  "y_in_dim": 768,
  "z_out_dim": 4096,
  "model_size": "9b",
  "sample_size": 16,
  "patch_size": 2,
  "latent_channels": 128,
  "weights": "model",
  "training_steps": 300000
}