didi-instruct-small / config.json
haoyangzheng's picture
Add generator/discriminator weights, config and README
2e9a21a
{
"_name_or_path": "haoyangzheng/didi-instruct-model",
"architectures": [
"DIT"
],
"auto_map": {
"AutoConfig": "modeling_dit.DITConfig",
"AutoModel": "modeling_dit.DIT"
},
"model_type": "dit-instruct",
"hidden_size": 768,
"cond_dim": 128,
"n_blocks": 12,
"n_heads": 12,
"dropout": 0.1,
"model_length": 1024,
"vocab_size": 50258,
"torch_dtype": "float32",
"transformers_version": "4.41.2"
}