{"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"
  }}