Upload SPIRAL step_00416
Browse files- README.md +6 -6
- chat_template.jinja +1 -0
- config.json +2 -2
- generation_config.json +1 -1
- model-00001-of-00004.safetensors +1 -1
- model-00002-of-00004.safetensors +1 -1
- model-00003-of-00004.safetensors +1 -1
- model-00004-of-00004.safetensors +1 -1
- model.safetensors.index.json +1 -0
- tokenizer_config.json +1 -2
README.md
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
---
|
| 2 |
-
base_model: OctoThinker-
|
| 3 |
license: apache-2.0
|
| 4 |
language:
|
| 5 |
- en
|
|
@@ -13,17 +13,17 @@ tags:
|
|
| 13 |
- multi-agent
|
| 14 |
---
|
| 15 |
|
| 16 |
-
# SPIRAL OctoThinker-
|
| 17 |
|
| 18 |
This model was trained using the SPIRAL (Self-Play Iterative Reinforcement learning for Adaptation and Learning) framework.
|
| 19 |
|
| 20 |
## Model Details
|
| 21 |
|
| 22 |
-
- **Base Model**: OctoAI/OctoThinker-
|
| 23 |
- **Training Framework**: SPIRAL
|
| 24 |
- **Checkpoint**: step_00416
|
| 25 |
-
- **Model Size**:
|
| 26 |
-
- **Training Date**: 2025-
|
| 27 |
|
| 28 |
## Training Configuration
|
| 29 |
|
|
@@ -45,7 +45,7 @@ The model was trained with self-play on multiple environments:
|
|
| 45 |
"TicTacToe-v0",
|
| 46 |
"SimpleNegotiation-v1"
|
| 47 |
],
|
| 48 |
-
"base_model": "OctoThinker-
|
| 49 |
"framework": "SPIRAL"
|
| 50 |
}
|
| 51 |
```
|
|
|
|
| 1 |
---
|
| 2 |
+
base_model: OctoAI/OctoThinker-3B
|
| 3 |
license: apache-2.0
|
| 4 |
language:
|
| 5 |
- en
|
|
|
|
| 13 |
- multi-agent
|
| 14 |
---
|
| 15 |
|
| 16 |
+
# SPIRAL OctoThinker-3B Multi-Agent Model
|
| 17 |
|
| 18 |
This model was trained using the SPIRAL (Self-Play Iterative Reinforcement learning for Adaptation and Learning) framework.
|
| 19 |
|
| 20 |
## Model Details
|
| 21 |
|
| 22 |
+
- **Base Model**: OctoAI/OctoThinker-3B
|
| 23 |
- **Training Framework**: SPIRAL
|
| 24 |
- **Checkpoint**: step_00416
|
| 25 |
+
- **Model Size**: 3B parameters
|
| 26 |
+
- **Training Date**: 2025-09-07
|
| 27 |
|
| 28 |
## Training Configuration
|
| 29 |
|
|
|
|
| 45 |
"TicTacToe-v0",
|
| 46 |
"SimpleNegotiation-v1"
|
| 47 |
],
|
| 48 |
+
"base_model": "OctoAI/OctoThinker-3B",
|
| 49 |
"framework": "SPIRAL"
|
| 50 |
}
|
| 51 |
```
|
chat_template.jinja
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{{ '<|begin_of_text|>' }}{% if messages[0]['role'] == 'system' %}{% set loop_messages = messages[1:] %}{% set system_message = messages[0]['content'] %}{% else %}{% set loop_messages = messages %}{% endif %}{% if system_message is defined %}{{ system_message }}{% endif %}{% for message in loop_messages %}{% set content = message['content'] %}{% if message['role'] == 'user' %}{{ content }}{% elif message['role'] == 'assistant' %}{{ content + '<|end_of_text|>' }}{% endif %}{% endfor %}
|
config.json
CHANGED
|
@@ -5,6 +5,7 @@
|
|
| 5 |
"attention_bias": false,
|
| 6 |
"attention_dropout": 0.0,
|
| 7 |
"bos_token_id": 128000,
|
|
|
|
| 8 |
"eos_token_id": 128001,
|
| 9 |
"head_dim": 128,
|
| 10 |
"hidden_act": "silu",
|
|
@@ -30,8 +31,7 @@
|
|
| 30 |
},
|
| 31 |
"rope_theta": 500000.0,
|
| 32 |
"tie_word_embeddings": false,
|
| 33 |
-
"
|
| 34 |
-
"transformers_version": "4.51.3",
|
| 35 |
"use_cache": false,
|
| 36 |
"vocab_size": 128256
|
| 37 |
}
|
|
|
|
| 5 |
"attention_bias": false,
|
| 6 |
"attention_dropout": 0.0,
|
| 7 |
"bos_token_id": 128000,
|
| 8 |
+
"dtype": "bfloat16",
|
| 9 |
"eos_token_id": 128001,
|
| 10 |
"head_dim": 128,
|
| 11 |
"hidden_act": "silu",
|
|
|
|
| 31 |
},
|
| 32 |
"rope_theta": 500000.0,
|
| 33 |
"tie_word_embeddings": false,
|
| 34 |
+
"transformers_version": "4.56.0",
|
|
|
|
| 35 |
"use_cache": false,
|
| 36 |
"vocab_size": 128256
|
| 37 |
}
|
generation_config.json
CHANGED
|
@@ -2,5 +2,5 @@
|
|
| 2 |
"_from_model_config": true,
|
| 3 |
"bos_token_id": 128000,
|
| 4 |
"eos_token_id": 128001,
|
| 5 |
-
"transformers_version": "4.
|
| 6 |
}
|
|
|
|
| 2 |
"_from_model_config": true,
|
| 3 |
"bos_token_id": 128000,
|
| 4 |
"eos_token_id": 128001,
|
| 5 |
+
"transformers_version": "4.56.0"
|
| 6 |
}
|
model-00001-of-00004.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 4976698672
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:909d8854901a285549b5df73d7b686635ac3e5314af943f0055ac89c28fe4bc8
|
| 3 |
size 4976698672
|
model-00002-of-00004.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 4999802720
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6f76f79b8cc0c2db8b969640eb0faecae7dac9ea525a36f94966eb276c2db166
|
| 3 |
size 4999802720
|
model-00003-of-00004.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 4915916176
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c7f4ef946a9c930046ec2a55651bbbdb1925224e3585caae9fb86afa056c7c04
|
| 3 |
size 4915916176
|
model-00004-of-00004.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1168138808
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d20572dff5401248dfb9d108b4bf2335e03a509f51fd70249dfa78cc9c82f50d
|
| 3 |
size 1168138808
|
model.safetensors.index.json
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"metadata": {
|
|
|
|
| 3 |
"total_size": 16060522496
|
| 4 |
},
|
| 5 |
"weight_map": {
|
|
|
|
| 1 |
{
|
| 2 |
"metadata": {
|
| 3 |
+
"total_parameters": 0,
|
| 4 |
"total_size": 16060522496
|
| 5 |
},
|
| 6 |
"weight_map": {
|
tokenizer_config.json
CHANGED
|
@@ -2050,7 +2050,6 @@
|
|
| 2050 |
}
|
| 2051 |
},
|
| 2052 |
"bos_token": "<|begin_of_text|>",
|
| 2053 |
-
"chat_template": "{{ '<|begin_of_text|>' }}{% if messages[0]['role'] == 'system' %}{% set loop_messages = messages[1:] %}{% set system_message = messages[0]['content'] %}{% else %}{% set loop_messages = messages %}{% endif %}{% if system_message is defined %}{{ system_message }}{% endif %}{% for message in loop_messages %}{% set content = message['content'] %}{% if message['role'] == 'user' %}{{ content }}{% elif message['role'] == 'assistant' %}{{ content + '<|end_of_text|>' }}{% endif %}{% endfor %}",
|
| 2054 |
"clean_up_tokenization_spaces": true,
|
| 2055 |
"eos_token": "<|end_of_text|>",
|
| 2056 |
"extra_special_tokens": {},
|
|
@@ -2062,5 +2061,5 @@
|
|
| 2062 |
"pad_token": "<|end_of_text|>",
|
| 2063 |
"padding_side": "left",
|
| 2064 |
"split_special_tokens": false,
|
| 2065 |
-
"tokenizer_class": "
|
| 2066 |
}
|
|
|
|
| 2050 |
}
|
| 2051 |
},
|
| 2052 |
"bos_token": "<|begin_of_text|>",
|
|
|
|
| 2053 |
"clean_up_tokenization_spaces": true,
|
| 2054 |
"eos_token": "<|end_of_text|>",
|
| 2055 |
"extra_special_tokens": {},
|
|
|
|
| 2061 |
"pad_token": "<|end_of_text|>",
|
| 2062 |
"padding_side": "left",
|
| 2063 |
"split_special_tokens": false,
|
| 2064 |
+
"tokenizer_class": "PreTrainedTokenizerFast"
|
| 2065 |
}
|