updating configuration

README.md

@@ -0,0 +1,12 @@
+---
+license: apache-2.0
+base_model: allenai/Olmo-3-32B-Think-DPO
+language:
+- en
+datasets:
+- allenai/Dolci-Think-RL
+library_name: transformers
+---
+
+# Model Details
+<img alt="Logo for Olmo 32B Think model" src="olmo-think.png" width="240px" style="margin-left:'auto' margin-right:'auto' display:'block'">

chat_template.jinja

@@ -13,5 +13,4 @@ You are Olmo, a helpful AI assistant built by Ai2. Your date cutoff is December
 ' }}{% else %}{{ eos_token }}{% endif %}{% elif message['role'] == 'environment' %}{{ '<|im_start|>environment
 ' + message['content'] + '<|im_end|>
 ' }}{% endif %}{% if loop.last and add_generation_prompt %}{{ '<|im_start|>assistant
-<think>' }}{% endif %}{% endfor %}
-
+<think>' }}{% endif %}{% endfor %}

generation_config.json

@@ -2,5 +2,8 @@
   "_from_model_config": true,
   "eos_token_id": 100257,
   "pad_token_id": 100277,
-  "transformers_version": "4.57.1"
-}
+  "transformers_version": "4.57.1",
+  "temperature": 0.6,
+  "top_p": 0.95,
+  "max_new_tokens": 32768
+}

model.safetensors.index.json

@@ -1,7 +1,7 @@
 {
   "metadata": {
-    "total_parameters": 1053696,
-    "total_size": 64467044352
+    "total_parameters": 32233522176,
+    "total_size": 128934088704
   },
   "weight_map": {
     "lm_head.weight": "model-00014-of-00014.safetensors",

tokenizer_config.json

@@ -185,5 +185,6 @@
   "model_max_length": 65536,
   "pad_token": "<|pad|>",
   "tokenizer_class": "GPT2Tokenizer",
-  "unk_token": "<|endoftext|>"
-}
+  "unk_token": "<|endoftext|>",
+  "chat_template": "{% set has_system = messages|selectattr('role', 'equalto', 'system')|list|length > 0 %}{% if not has_system %}{{ '<|im_start|>system\nYou are Olmo, a helpful AI assistant built by Ai2. Your date cutoff is December 2024, and your model weights are available at https://huggingface.co/allenai.<|im_end|>\n' }}{% endif %}{% for message in messages %}{% if message['role'] == 'system' %}{{ '<|im_start|>system\n' + message['content'] }}{% if message.get('functions', none) is not none %}{{ ' <functions>' + message['functions'] + '</functions><|im_end|>\n' }}{% else %}{{ ' You do not currently have access to any functions. <functions></functions><|im_end|>\n' }}{% endif %}{% elif message['role'] == 'user' %}{% if message.get('functions', none) is not none %}{{ '<|im_start|>user\n' + message['content'] + '\n' + '<functions>' + message['functions'] + '</functions><|im_end|>\n' }}{% else %}{{ '<|im_start|>user\n' + message['content'] + '<|im_end|>\n' }}{% endif %}{% elif message['role'] == 'assistant' %}{{ '<|im_start|>assistant\n' }}{% if message.get('content', none) is not none %}{{ message['content'] }}{% endif %}{% if message.get('function_calls', none) is not none %}{{ '<function_calls>' + message['function_calls'] + '</function_calls>' }}{% endif %}{% if not loop.last %}{{ '<|im_end|>' + '\n' }}{% else %}{{ eos_token }}{% endif %}{% elif message['role'] == 'environment' %}{{ '<|im_start|>environment\n' + message['content'] + '<|im_end|>\n' }}{% endif %}{% if loop.last and add_generation_prompt %}{{ '<|im_start|>assistant\n<think>' }}{% endif %}{% endfor %}"
+}