| { | |
| "_class_name": "PrismDiT", | |
| "model_type": "prism-dit", | |
| "architecture": "PrismDiT", | |
| "in_channels": 16, | |
| "out_channels": 16, | |
| "patch_size": 2, | |
| "hidden_size": 1152, | |
| "depth": 28, | |
| "num_heads": 16, | |
| "mlp_ratio": 4.0, | |
| "context_dim": 1024, | |
| "diffusion_type": "flow-matching", | |
| "scheduler": "PrismFlowScheduler", | |
| "recommended_sampler": "Euler", | |
| "recommended_steps": "12-25", | |
| "recommended_cfg": "5.0-6.5", | |
| "torch_dtype": "float16", | |
| "version": "1.0.0" | |
| } | |