Update config.json
Browse files- config.json +3 -1
config.json
CHANGED
|
@@ -1 +1,3 @@
|
|
| 1 |
-
{"vocab_size": 57372, "max_seq_len": 99, "max_time_steps": 10, "embed_dim": 128, "n_layers": 4, "n_heads": 4
|
|
|
|
|
|
|
|
|
| 1 |
+
{"vocab_size": 57372, "max_seq_len": 99, "max_time_steps": 10, "embed_dim": 128, "n_layers": 4, "n_heads": 4,"auto_map": {
|
| 2 |
+
"AutoModel": "modeling_diffusion.DiffusionTextModel"
|
| 3 |
+
}}
|