simonycl commited on
Commit
c931b28
·
verified ·
1 Parent(s): 566e14c

Upload SPIRAL step_00416

Browse files
README.md CHANGED
@@ -1,5 +1,5 @@
1
  ---
2
- base_model: OctoThinker-8B
3
  license: apache-2.0
4
  language:
5
  - en
@@ -13,17 +13,17 @@ tags:
13
  - multi-agent
14
  ---
15
 
16
- # SPIRAL OctoThinker-8B 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-8B
23
  - **Training Framework**: SPIRAL
24
  - **Checkpoint**: step_00416
25
- - **Model Size**: 8B parameters
26
- - **Training Date**: 2025-08-27
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-8B",
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
- "torch_dtype": "bfloat16",
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.51.3"
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:4c12e766d11fd9a640099b2bc670cb55ebcd664de4169dc50feace04ec82707f
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:4a950d37fc4859efe19d7494e36d91dc0e5e30edb45489ed4d6d095959fd07e9
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:d5dbf35ad64748568bfee50d6505fa0b531abc7bcebd3e84af13fdfdfa3ec3d9
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:95a7632bf5130e403e7c38f1e30218f9628cbf8e04a673522f7f516bc3503518
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": "PreTrainedTokenizer"
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
  }