diffusion-text-demo / config.json
yasserrmd's picture
Update config.json
d60b639 verified
raw
history blame contribute delete
184 Bytes
{"vocab_size": 57372, "max_seq_len": 99, "max_time_steps": 10, "embed_dim": 128, "n_layers": 4, "n_heads": 4,"auto_map": {
"AutoModel": "modeling_diffusion.DiffusionTextModel"
}}