Upload folder using huggingface_hub

chat_template.jinja

@@ -0,0 +1,45 @@
+
+{%- for message in messages %}
+    {#-- Validate role is a stringified integer --#}
+    {%- if not message['role'] is string or not message['role'].isdigit() %}
+        {{- raise_exception("The role must be an integer or a stringified integer (e.g. '0') designating the speaker id") }}
+    {%- endif %}
+
+    {#-- Validate content is a list --#}
+    {%- set content = message['content'] %}
+    {%- if content is not iterable or content is string %}
+        {{- raise_exception("The content must be a list") }}
+    {%- endif %}
+
+    {#-- Collect content types --#}
+    {%- set content_types = content | map(attribute='type') | list %}
+    {%- set is_last = loop.last %}
+
+    {#-- Last message validation --#}
+    {%- if is_last %}
+        {%- if 'text' not in content_types %}
+            {{- raise_exception("The last message must include one item of type 'text'") }}
+        {%- elif (content_types | select('equalto', 'text') | list | length > 1) or (content_types | select('equalto', 'audio') | list | length > 1) %}
+            {{- raise_exception("At most two items are allowed in the last message: one 'text' and one 'audio'") }}
+        {%- endif %}
+
+    {#-- All other messages validation --#}
+    {%- else %}
+        {%- if content_types | select('equalto', 'text') | list | length != 1
+              or content_types | select('equalto', 'audio') | list | length != 1 %}
+            {{- raise_exception("Each message (except the last) must contain exactly one 'text' and one 'audio' item") }}
+        {%- elif content_types | reject('in', ['text', 'audio']) | list | length > 0 %}
+            {{- raise_exception("Only 'text' and 'audio' types are allowed in content") }}
+        {%- endif %}
+    {%- endif %}
+{%- endfor %}
+
+{%- for message in messages %}
+    {{- bos_token }}
+    {{- '[' + message['role'] + ']' }}
+    {{- message['content'][0]['text'] }}
+    {{- eos_token }}
+    {%- if message['content']|length > 1 %}
+        {{- '<|AUDIO|><|audio_eos|>' }}
+    {%- endif %}
+{%- endfor %}

config.json

@@ -1,6 +1,6 @@
 {
   "architectures": [
-    "CsmForCausalLM"
+    "CsmForConditionalGeneration"
   ],
   "attention_bias": false,
   "attention_dropout": 0.0,
@@ -114,7 +114,7 @@
   "tie_codebooks_embeddings": true,
   "tie_word_embeddings": false,
   "torch_dtype": "float32",
-  "transformers_version": "4.51.0.dev0",
+  "transformers_version": "4.52.0.dev0",
   "use_cache": true,
   "vocab_size": 2051
 }

generation_config.json

@@ -1,6 +1,10 @@
 {
-  "_from_model_config": true,
   "bos_token_id": 128000,
+  "depth_decoder_do_sample": true,
+  "depth_decoder_temperature": 0.9,
+  "depth_decoder_top_k": 50,
+  "do_sample": true,
   "pad_token_id": 128002,
-  "transformers_version": "4.51.0.dev0"
+  "temperature": 0.9,
+  "transformers_version": "4.52.0.dev0"
 }