EthanZyh's picture
put _hf inside, modify config correspondingly
701ab29
raw
history blame
304 Bytes
{
"architectures": [
"DiffusionText2World"
],
"auto_map": {
"AutoConfig": "cosmos1/models/diffusion/inference/text2world_hf.DiffusionText2WorldConfig",
"AutoModel": "cosmos1/models/diffusion/inference/text2world_hf.DiffusionText2World"
},
"model_type": "AutoModel"
}