config.json

{
  "architectures": [
    "SupraDiT"
  ],
  "model_type": "SupraDiT",
  "auto_map": {
    "AutoConfig": "configuration_my_model.SupraDiTConfig",
    "AutoModel": "modeling_my_model.SupraDiT"
  },
  "torch_dtype": "bfloat16"
}