Upload model

config.json

@@ -2,6 +2,7 @@
   "architectures": [
     "DiaModel"
   ],
+  "bos_token_id": 1026,
   "decoder_config": {
     "_attn_implementation_autoset": true,
     "cross_head_dim": 128,
@@ -20,8 +21,20 @@
     "num_key_value_heads": 4,
     "rope_max_timescale": 10000,
     "rope_min_timescale": 1,
+    "torch_dtype": "float32",
     "vocab_size": 1028
   },
+  "delay_pattern": [
+    0,
+    8,
+    9,
+    10,
+    11,
+    12,
+    13,
+    14,
+    15
+  ],
   "encoder_config": {
     "_attn_implementation_autoset": true,
     "dropout": 0,
@@ -36,12 +49,14 @@
     "num_key_value_heads": 16,
     "rope_max_timescale": 10000,
     "rope_min_timescale": 1,
+    "torch_dtype": "float32",
     "vocab_size": 256
   },
+  "eos_token_id": 1024,
   "is_encoder_decoder": true,
   "model_type": "dia",
   "norm_eps": 1e-05,
-  "pad_token_id": 0,
+  "pad_token_id": 1025,
   "torch_dtype": "float32",
   "transformers_version": "4.52.0.dev0"
 }

generation_config.json

@@ -0,0 +1,7 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 1026,
+  "eos_token_id": 1024,
+  "pad_token_id": 1025,
+  "transformers_version": "4.52.0.dev0"
+}

model-00001-of-00002.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ce722238a3a86ced278ff8dbe5ecc56c2e3bb61521fc3b8ee992986d076e0105
+oid sha256:2c7d787ff2bc1f85dcdb74e4ccfdc9dd7a59c03bcd5df0f3b534f17894eaed77
 size 4993044448

model-00002-of-00002.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b3efc0902a2ce0f2200bd53a04a549aa6b9bd7d6053b07df8be394a6ad0de672
+oid sha256:5b9b3eed74c98c55f0b4f3ba3f8c3ccb65716f9f3c48bf1bda72f1433b95f27a
 size 1451637168