{ "_class_name": "SiTTransformer2DModel", "_diffusers_version": "0.36.0", "class_dropout_prob": 0.1, "depth": 12, "hidden_size": 768, "in_channels": 4, "input_size": 32, "learn_sigma": true, "mlp_ratio": 4.0, "num_classes": 1000, "num_heads": 12, "patch_size": 2 }