Upload SPIRAL step_00288
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_00288
|
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_00288
|
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:8ca0d7431c9d9d21fd8e75a3ff8181eb7e5145a73349e6bde16bc2402319437f
|
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:fdd7891ee5dc263aa55bc917f96fb17fb11b9334444990f272ae45921cc85dba
|
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:79b4eee55050894c3591792baeae8fb2fec7ce6dd6f17f409d10e4dfd090e3a4
|
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:9d99b8614d0696100fe8719ab80a4dd60f04b9ae7ce1f1c278b0f64b9b68d707
|
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 |
}
|