zeromodels/stable-diffusion-3.5-large

1{2  "library_name": "zeromodels",3  "zeromodels_version": "1.3.3",4  "model_module": "zeromodels.models.stable_diffusion_3_5",5  "model_class": "StableDiffusion3_5Model",6  "variant": "stable-diffusion-3.5-large",7  "weights": "model.weights.json",8  "schema_version": 2,9  "weight_dtype": "float16",10  "weight_dtype_overrides": {11    "vae_encoder": "float32",12    "vae_decoder": "float32"13  },14  "model_type": "stable_diffusion_3_5",15  "transformer_config": {16    "sample_size": 128,17    "patch_size": 2,18    "in_channels": 16,19    "out_channels": 16,20    "num_layers": 38,21    "attention_head_dim": 64,22    "num_attention_heads": 38,23    "joint_attention_dim": 4096,24    "caption_projection_dim": 2432,25    "pooled_projection_dim": 2048,26    "pos_embed_max_size": 192,27    "qk_norm": "rms_norm",28    "dual_attention_layers": [],29    "text_seq_len": 33330  },31  "vae_config": {32    "in_channels": 3,33    "out_channels": 3,34    "latent_channels": 16,35    "block_out_channels": [36      128,37      256,38      512,39      51240    ],41    "layers_per_block": 2,42    "norm_num_groups": 32,43    "sample_size": 1024,44    "scaling_factor": 1.5305,45    "force_upcast": true,46    "shift_factor": 0.0609,47    "use_quant_conv": false,48    "use_post_quant_conv": false49  },50  "text_config": {51    "hidden_dim": 768,52    "num_heads": 12,53    "num_layers": 12,54    "mlp_ratio": 4.0,55    "vocab_size": 49408,56    "max_seq_len": 77,57    "projection_dim": 768,58    "hidden_act": "quick_gelu"59  },60  "text_config_2": {61    "hidden_dim": 1280,62    "num_heads": 20,63    "num_layers": 32,64    "mlp_ratio": 4.0,65    "vocab_size": 49408,66    "max_seq_len": 77,67    "projection_dim": 1280,68    "hidden_act": "gelu"69  },70  "hidden_act": "quick_gelu",71  "layer_norm_eps": 1e-05,72  "bos_token_id": 49406,73  "eos_token_id": 49407,74  "pad_token_id": 49407,75  "pad_token_id_2": 0,76  "max_sequence_length": 256,77  "eos_token_id_3": 1,78  "pad_token_id_3": 0,79  "scheduler_config": {80    "_class_name": "FlowMatchEulerDiscreteScheduler",81    "num_train_timesteps": 1000,82    "shift": 3.083  },84  "generate_args": {85    "num_inference_steps": 28,86    "guidance_scale": 3.587  }88}