初始化项目,由ModelHub XC社区提供模型

Model: tool-genesis/Tool-Genesis-Qwen3-8B-SFT
Source: Original Platform
This commit is contained in:
ModelHub XC
2026-06-15 17:28:30 +08:00
commit 9bfee43b56
22 changed files with 153488 additions and 0 deletions

39
.gitattributes vendored Normal file
View File

@@ -0,0 +1,39 @@
*.7z filter=lfs diff=lfs merge=lfs -text
*.arrow filter=lfs diff=lfs merge=lfs -text
*.bin filter=lfs diff=lfs merge=lfs -text
*.bz2 filter=lfs diff=lfs merge=lfs -text
*.ckpt filter=lfs diff=lfs merge=lfs -text
*.ftz filter=lfs diff=lfs merge=lfs -text
*.gz filter=lfs diff=lfs merge=lfs -text
*.h5 filter=lfs diff=lfs merge=lfs -text
*.joblib filter=lfs diff=lfs merge=lfs -text
*.lfs.* filter=lfs diff=lfs merge=lfs -text
*.mlmodel filter=lfs diff=lfs merge=lfs -text
*.model filter=lfs diff=lfs merge=lfs -text
*.msgpack filter=lfs diff=lfs merge=lfs -text
*.npy filter=lfs diff=lfs merge=lfs -text
*.npz filter=lfs diff=lfs merge=lfs -text
*.onnx filter=lfs diff=lfs merge=lfs -text
*.ot filter=lfs diff=lfs merge=lfs -text
*.parquet filter=lfs diff=lfs merge=lfs -text
*.pb filter=lfs diff=lfs merge=lfs -text
*.pickle filter=lfs diff=lfs merge=lfs -text
*.pkl filter=lfs diff=lfs merge=lfs -text
*.pt filter=lfs diff=lfs merge=lfs -text
*.pth filter=lfs diff=lfs merge=lfs -text
*.rar filter=lfs diff=lfs merge=lfs -text
*.safetensors filter=lfs diff=lfs merge=lfs -text
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
*.tar.* filter=lfs diff=lfs merge=lfs -text
*.tar filter=lfs diff=lfs merge=lfs -text
*.tflite filter=lfs diff=lfs merge=lfs -text
*.tgz filter=lfs diff=lfs merge=lfs -text
*.wasm filter=lfs diff=lfs merge=lfs -text
*.xz filter=lfs diff=lfs merge=lfs -text
*.zip filter=lfs diff=lfs merge=lfs -text
*.zst filter=lfs diff=lfs merge=lfs -text
*tfevents* filter=lfs diff=lfs merge=lfs -text
tokenizer.json filter=lfs diff=lfs merge=lfs -text
checkpoint-117/tokenizer.json filter=lfs diff=lfs merge=lfs -text
checkpoint-39/tokenizer.json filter=lfs diff=lfs merge=lfs -text
checkpoint-78/tokenizer.json filter=lfs diff=lfs merge=lfs -text

171
README.md Normal file
View File

@@ -0,0 +1,171 @@
---
language:
- en
- zh
license: apache-2.0
library_name: transformers
base_model: Qwen/Qwen3-8B
tags:
- qwen3
- tool-use
- mcp
- code-generation
- sft
datasets:
- tool-genesis/Tool-Genesis-Benchmark
pipeline_tag: text-generation
model-index:
- name: Tool-Genesis-Qwen3-8B-SFT
results:
- task:
type: text-generation
name: MCP Server Generation (Direct)
dataset:
name: Tool-Genesis Benchmark
type: tool-genesis/Tool-Genesis-Benchmark
metrics:
- type: compliance
value: 0.826
name: L1 Compliance
- type: launch_rate
value: 0.047
name: L1 Launch Rate
- type: schema_f1
value: 0.046
name: L2 Schema F1
- type: ut_soft
value: 0.017
name: L2 UT Soft
---
# Tool-Genesis-Qwen3-8B-SFT
A fine-tuned [Qwen3-8B](https://huggingface.co/Qwen/Qwen3-8B) model for autonomous MCP (Model Context Protocol) tool server generation. Given a natural language scenario description, the model generates a complete, runnable MCP server with tool schemas and implementation code.
## Model Details
| Property | Value |
|---|---|
| Base model | [Qwen/Qwen3-8B](https://huggingface.co/Qwen/Qwen3-8B) |
| Architecture | Qwen3ForCausalLM |
| Parameters | 8B |
| Hidden size | 4096 |
| Layers | 36 |
| Attention heads | 32 |
| Context length | 131,072 tokens |
| Training method | Full-parameter SFT |
| Training epochs | 3 |
| Training steps | 117 |
| Training loss | 0.522 |
| Training data | ~2,500 samples |
## Training
The model was fine-tuned on curated MCP server generation examples from the Tool-Genesis benchmark. Each training sample consists of:
- **Input**: A natural language scenario description specifying what the MCP server should do
- **Output**: A complete Python MCP server implementation using the FastMCP framework
### Training Configuration
- **Epochs**: 3
- **Total steps**: 117 (~39 steps/epoch)
- **Final training loss**: 0.522
- **Training runtime**: ~4.6 hours
### Loss Curve
| Step | Loss |
|---|---|
| 1 | 0.763 |
| 10 | 0.690 |
| 20 | 0.641 |
| 39 (epoch 1) | 0.539 |
| 60 | 0.434 |
| 78 (epoch 2) | 0.436 |
| 100 | 0.420 |
| 117 (epoch 3) | 0.522 |
## Benchmark Results
Evaluated on the [Tool-Genesis Benchmark](https://huggingface.co/datasets/tool-genesis/Tool-Genesis-Benchmark) (86 MCP servers, 4-level evaluation).
### Direct Generation (single-call, no agent loop)
| Model | L1 Compliance | L1 Launch | L2 Schema F1 | L2 UT Soft |
|---|---|---|---|---|
| Qwen3-8B (base) | 0.686 | 0.012 | 0.011 | 0.001 |
| **Qwen3-8B-SFT (ours)** | **0.826** | **0.047** | **0.046** | **0.017** |
| Qwen3-235B | 0.874 | 0.333 | 0.316 | 0.142 |
| GPT-4.1 | 0.881 | 0.738 | 0.691 | 0.267 |
| GPT-5.1 | 0.855 | 0.759 | 0.713 | 0.291 |
**SFT gains over base Qwen3-8B (Direct):**
- L1 Compliance: +14.0% (0.686 → 0.826)
- L1 Launch: +3.5% (0.012 → 0.047)
- L2 Schema F1: +3.5% (0.011 → 0.046)
- L2 UT Soft: +1.6% (0.001 → 0.017)
### With Coder-Agent (multi-turn with sandbox)
| Model | L1 Compliance | L1 Launch | L2 Schema F1 | L2 UT Soft |
|---|---|---|---|---|
| Qwen3-8B (base, coder-agent) | 0.776 | 0.694 | 0.653 | 0.246 |
| Qwen3-235B (coder-agent) | 0.868 | 0.971 | 0.914 | 0.459 |
| GPT-4.1 (coder-agent) | 0.884 | 0.756 | 0.691 | 0.288 |
| GPT-5.1 (coder-agent) | 0.906 | 0.941 | 0.877 | 0.426 |
> Note: The coder-agent strategy dramatically improves all models by providing an iterative sandbox-based coding loop. The SFT model has not yet been evaluated with the coder-agent strategy.
## Usage
```python
from transformers import AutoModelForCausalLM, AutoTokenizer
model_name = "tool-genesis/Tool-Genesis-Qwen3-8B-SFT"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto", device_map="auto")
prompt = """You are a developer building MCP tool servers in Python.
Build a complete MCP server for the following scenario:
A weather information service that provides current weather data,
forecasts, and weather alerts for any location worldwide.
Output only the Python source code using the FastMCP framework."""
inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
outputs = model.generate(**inputs, max_new_tokens=4096, temperature=0.2)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
```
## Evaluation Protocol
The Tool-Genesis benchmark evaluates generated MCP servers across four levels:
| Level | What it tests |
|---|---|
| **L1: Protocol Compliance** | JSON format validity and server launch success |
| **L2: Semantic Correctness** | Tool schema matching (F1) and unit test pass rate |
| **L3: Capability Boundary** | No unauthorized capabilities or dangerous extra tools |
| **L4: Task Utility** | Downstream task completion using generated tools |
## Links
- **Benchmark Dataset**: [tool-genesis/Tool-Genesis-Benchmark](https://huggingface.co/datasets/tool-genesis/Tool-Genesis-Benchmark)
- **Code**: [github.com/Tool-Genesis/Tool-Genesis](https://github.com/Tool-Genesis/Tool-Genesis)
## Citation
```bibtex
@misc{tool_genesis_2025,
title={Tool-Genesis: A Task-Driven Tool Creation Benchmark for Self-Evolving Language Agent},
author={Xia, Bowei and Hu, Mengkang and Wang, Shijian and Jin, Jiarui and Jiao, Wenxiang and Lu, Yuan and Li, Kexin and Luo, Ping},
year={2025},
note={Project page: https://tool-genesis.github.io}
}
```
## License
Apache 2.0

28
added_tokens.json Normal file
View File

@@ -0,0 +1,28 @@
{
"</think>": 151668,
"</tool_call>": 151658,
"</tool_response>": 151666,
"<think>": 151667,
"<tool_call>": 151657,
"<tool_response>": 151665,
"<|box_end|>": 151649,
"<|box_start|>": 151648,
"<|endoftext|>": 151643,
"<|file_sep|>": 151664,
"<|fim_middle|>": 151660,
"<|fim_pad|>": 151662,
"<|fim_prefix|>": 151659,
"<|fim_suffix|>": 151661,
"<|im_end|>": 151645,
"<|im_start|>": 151644,
"<|image_pad|>": 151655,
"<|object_ref_end|>": 151647,
"<|object_ref_start|>": 151646,
"<|quad_end|>": 151651,
"<|quad_start|>": 151650,
"<|repo_name|>": 151663,
"<|video_pad|>": 151656,
"<|vision_end|>": 151653,
"<|vision_pad|>": 151654,
"<|vision_start|>": 151652
}

8
all_results.json Normal file
View File

@@ -0,0 +1,8 @@
{
"epoch": 3.0,
"total_flos": 3.3860293178247086e+18,
"train_loss": 0.5216725878735893,
"train_runtime": 16633.6833,
"train_samples_per_second": 0.449,
"train_steps_per_second": 0.007
}

89
chat_template.jinja Normal file
View File

@@ -0,0 +1,89 @@
{%- if tools %}
{{- '<|im_start|>system\n' }}
{%- if messages[0].role == 'system' %}
{{- messages[0].content + '\n\n' }}
{%- endif %}
{{- "# Tools\n\nYou may call one or more functions to assist with the user query.\n\nYou are provided with function signatures within <tools></tools> XML tags:\n<tools>" }}
{%- for tool in tools %}
{{- "\n" }}
{{- tool | tojson }}
{%- endfor %}
{{- "\n</tools>\n\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\n<tool_call>\n{\"name\": <function-name>, \"arguments\": <args-json-object>}\n</tool_call><|im_end|>\n" }}
{%- else %}
{%- if messages[0].role == 'system' %}
{{- '<|im_start|>system\n' + messages[0].content + '<|im_end|>\n' }}
{%- endif %}
{%- endif %}
{%- set ns = namespace(multi_step_tool=true, last_query_index=messages|length - 1) %}
{%- for message in messages[::-1] %}
{%- set index = (messages|length - 1) - loop.index0 %}
{%- if ns.multi_step_tool and message.role == "user" and message.content is string and not(message.content.startswith('<tool_response>') and message.content.endswith('</tool_response>')) %}
{%- set ns.multi_step_tool = false %}
{%- set ns.last_query_index = index %}
{%- endif %}
{%- endfor %}
{%- for message in messages %}
{%- if message.content is string %}
{%- set content = message.content %}
{%- else %}
{%- set content = '' %}
{%- endif %}
{%- if (message.role == "user") or (message.role == "system" and not loop.first) %}
{{- '<|im_start|>' + message.role + '\n' + content + '<|im_end|>' + '\n' }}
{%- elif message.role == "assistant" %}
{%- set reasoning_content = '' %}
{%- if message.reasoning_content is string %}
{%- set reasoning_content = message.reasoning_content %}
{%- else %}
{%- if '</think>' in content %}
{%- set reasoning_content = content.split('</think>')[0].rstrip('\n').split('<think>')[-1].lstrip('\n') %}
{%- set content = content.split('</think>')[-1].lstrip('\n') %}
{%- endif %}
{%- endif %}
{%- if loop.index0 > ns.last_query_index %}
{%- if loop.last or (not loop.last and reasoning_content) %}
{{- '<|im_start|>' + message.role + '\n<think>\n' + reasoning_content.strip('\n') + '\n</think>\n\n' + content.lstrip('\n') }}
{%- else %}
{{- '<|im_start|>' + message.role + '\n' + content }}
{%- endif %}
{%- else %}
{{- '<|im_start|>' + message.role + '\n' + content }}
{%- endif %}
{%- if message.tool_calls %}
{%- for tool_call in message.tool_calls %}
{%- if (loop.first and content) or (not loop.first) %}
{{- '\n' }}
{%- endif %}
{%- if tool_call.function %}
{%- set tool_call = tool_call.function %}
{%- endif %}
{{- '<tool_call>\n{"name": "' }}
{{- tool_call.name }}
{{- '", "arguments": ' }}
{%- if tool_call.arguments is string %}
{{- tool_call.arguments }}
{%- else %}
{{- tool_call.arguments | tojson }}
{%- endif %}
{{- '}\n</tool_call>' }}
{%- endfor %}
{%- endif %}
{{- '<|im_end|>\n' }}
{%- elif message.role == "tool" %}
{%- if loop.first or (messages[loop.index0 - 1].role != "tool") %}
{{- '<|im_start|>user' }}
{%- endif %}
{{- '\n<tool_response>\n' }}
{{- content }}
{{- '\n</tool_response>' }}
{%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
{{- '<|im_end|>\n' }}
{%- endif %}
{%- endif %}
{%- endfor %}
{%- if add_generation_prompt %}
{{- '<|im_start|>assistant\n' }}
{%- if enable_thinking is defined and enable_thinking is false %}
{{- '<think>\n\n</think>\n\n' }}
{%- endif %}
{%- endif %}

68
config.json Normal file
View File

@@ -0,0 +1,68 @@
{
"architectures": [
"Qwen3ForCausalLM"
],
"attention_bias": false,
"attention_dropout": 0.0,
"dtype": "bfloat16",
"eos_token_id": 151645,
"head_dim": 128,
"hidden_act": "silu",
"hidden_size": 4096,
"initializer_range": 0.02,
"intermediate_size": 12288,
"layer_types": [
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention"
],
"max_position_embeddings": 40960,
"max_window_layers": 36,
"model_type": "qwen3",
"num_attention_heads": 32,
"num_hidden_layers": 36,
"num_key_value_heads": 8,
"pad_token_id": 151643,
"rms_norm_eps": 1e-06,
"rope_scaling": null,
"rope_theta": 1000000,
"sliding_window": null,
"tie_word_embeddings": false,
"transformers_version": "4.57.3",
"use_cache": false,
"use_sliding_window": false,
"vocab_size": 151936
}

12
generation_config.json Normal file
View File

@@ -0,0 +1,12 @@
{
"do_sample": true,
"eos_token_id": [
151645,
151643
],
"pad_token_id": 151643,
"temperature": 0.6,
"top_k": 20,
"top_p": 0.95,
"transformers_version": "4.57.3"
}

151388
merges.txt Normal file

File diff suppressed because it is too large Load Diff

View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:c6c28c1669d3680e051f05928e0208606e1bdceb2d0df08cd6de9a2bb30963dd
size 4902257696

View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:11d0b2db57b543153e0a35ee71439ecae65563487c0a256eae22feec011c7597
size 4915960368

View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:f351bc28d98749c1d5ae0a3645af92f1f14614648f29341fa78489bcc5cdffa8
size 4983068496

View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:39dad5c027e0fd9704efb37c6963b7f3e9f2444db99c6574e2dfc69ebffde35e
size 1580230264

View File

@@ -0,0 +1,407 @@
{
"metadata": {
"total_parameters": 8190735360,
"total_size": 16381470720
},
"weight_map": {
"lm_head.weight": "model-00004-of-00004.safetensors",
"model.embed_tokens.weight": "model-00001-of-00004.safetensors",
"model.layers.0.input_layernorm.weight": "model-00001-of-00004.safetensors",
"model.layers.0.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.0.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.0.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.0.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
"model.layers.0.self_attn.k_norm.weight": "model-00001-of-00004.safetensors",
"model.layers.0.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.0.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.0.self_attn.q_norm.weight": "model-00001-of-00004.safetensors",
"model.layers.0.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.0.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.1.input_layernorm.weight": "model-00001-of-00004.safetensors",
"model.layers.1.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.1.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.1.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.1.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
"model.layers.1.self_attn.k_norm.weight": "model-00001-of-00004.safetensors",
"model.layers.1.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.1.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.1.self_attn.q_norm.weight": "model-00001-of-00004.safetensors",
"model.layers.1.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.1.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.10.input_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.10.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.10.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.10.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.10.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.10.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.10.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.10.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.10.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.10.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.10.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.11.input_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.11.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.11.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.11.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.11.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.11.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.11.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.11.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.11.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.11.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.11.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.12.input_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.12.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.12.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.12.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.12.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.12.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.12.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.12.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.12.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.12.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.12.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.13.input_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.13.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.13.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.13.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.13.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.13.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.13.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.13.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.13.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.13.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.13.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.14.input_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.14.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.14.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.14.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.14.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.14.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.14.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.14.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.14.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.14.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.14.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.15.input_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.15.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.15.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.15.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.15.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.15.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.15.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.15.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.15.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.15.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.15.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.16.input_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.16.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.16.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.16.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.16.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.16.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.16.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.16.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.16.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.16.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.16.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.17.input_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.17.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.17.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.17.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.17.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.17.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.17.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.17.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.17.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.17.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.17.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.18.input_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.18.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.18.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.18.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.18.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.18.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.18.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.18.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.18.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.18.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.18.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.19.input_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.19.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.19.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.19.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.19.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.19.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.19.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.19.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.19.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.19.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.19.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.2.input_layernorm.weight": "model-00001-of-00004.safetensors",
"model.layers.2.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.2.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.2.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.2.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
"model.layers.2.self_attn.k_norm.weight": "model-00001-of-00004.safetensors",
"model.layers.2.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.2.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.2.self_attn.q_norm.weight": "model-00001-of-00004.safetensors",
"model.layers.2.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.2.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.20.input_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.20.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.20.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.20.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.20.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.20.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.20.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.20.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.20.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.20.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.20.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.21.input_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.21.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.21.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.21.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.21.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.21.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.21.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.21.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.21.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.21.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.21.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.22.input_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.22.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.22.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.22.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.22.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.22.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.22.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.22.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.22.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
"model.layers.22.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.22.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.23.input_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.23.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.23.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.23.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.23.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.23.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.23.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.23.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.23.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.23.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.23.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.24.input_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.24.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.24.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.24.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.24.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.24.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.24.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.24.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.24.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.24.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.24.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.25.input_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.25.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.25.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.25.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.25.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.25.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.25.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.25.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.25.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.25.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.25.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.26.input_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.26.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.26.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.26.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.26.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.26.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.26.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.26.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.26.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.26.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.26.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.27.input_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.27.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.27.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.27.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.27.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.27.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.27.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.27.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.27.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.27.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.27.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.28.input_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.28.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.28.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.28.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.28.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.28.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.28.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.28.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.28.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.28.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.28.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.29.input_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.29.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.29.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.29.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.29.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.29.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.29.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.29.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.29.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.29.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.29.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.3.input_layernorm.weight": "model-00001-of-00004.safetensors",
"model.layers.3.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.3.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.3.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.3.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
"model.layers.3.self_attn.k_norm.weight": "model-00001-of-00004.safetensors",
"model.layers.3.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.3.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.3.self_attn.q_norm.weight": "model-00001-of-00004.safetensors",
"model.layers.3.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.3.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.30.input_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.30.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.30.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.30.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.30.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.30.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.30.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.30.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.30.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.30.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.30.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.31.input_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.31.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.31.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.31.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.31.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.31.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.31.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.31.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.31.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.31.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.31.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.32.input_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.32.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.32.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.32.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.32.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.32.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.32.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.32.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.32.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.32.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.32.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.33.input_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.33.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.33.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.33.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.33.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.33.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.33.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.33.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.33.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.33.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.33.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.34.input_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.34.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.34.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.34.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.34.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
"model.layers.34.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.34.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.34.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.34.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
"model.layers.34.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.34.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.35.input_layernorm.weight": "model-00004-of-00004.safetensors",
"model.layers.35.mlp.down_proj.weight": "model-00004-of-00004.safetensors",
"model.layers.35.mlp.gate_proj.weight": "model-00004-of-00004.safetensors",
"model.layers.35.mlp.up_proj.weight": "model-00004-of-00004.safetensors",
"model.layers.35.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
"model.layers.35.self_attn.k_norm.weight": "model-00004-of-00004.safetensors",
"model.layers.35.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.35.self_attn.o_proj.weight": "model-00004-of-00004.safetensors",
"model.layers.35.self_attn.q_norm.weight": "model-00004-of-00004.safetensors",
"model.layers.35.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.35.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
"model.layers.4.input_layernorm.weight": "model-00001-of-00004.safetensors",
"model.layers.4.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.4.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.4.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.4.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
"model.layers.4.self_attn.k_norm.weight": "model-00001-of-00004.safetensors",
"model.layers.4.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.4.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.4.self_attn.q_norm.weight": "model-00001-of-00004.safetensors",
"model.layers.4.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.4.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.5.input_layernorm.weight": "model-00001-of-00004.safetensors",
"model.layers.5.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.5.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.5.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.5.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
"model.layers.5.self_attn.k_norm.weight": "model-00001-of-00004.safetensors",
"model.layers.5.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.5.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.5.self_attn.q_norm.weight": "model-00001-of-00004.safetensors",
"model.layers.5.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.5.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.6.input_layernorm.weight": "model-00001-of-00004.safetensors",
"model.layers.6.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.6.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.6.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.6.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
"model.layers.6.self_attn.k_norm.weight": "model-00001-of-00004.safetensors",
"model.layers.6.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.6.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.6.self_attn.q_norm.weight": "model-00001-of-00004.safetensors",
"model.layers.6.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.6.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.7.input_layernorm.weight": "model-00001-of-00004.safetensors",
"model.layers.7.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.7.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.7.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.7.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
"model.layers.7.self_attn.k_norm.weight": "model-00001-of-00004.safetensors",
"model.layers.7.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.7.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.7.self_attn.q_norm.weight": "model-00001-of-00004.safetensors",
"model.layers.7.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.7.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.8.input_layernorm.weight": "model-00001-of-00004.safetensors",
"model.layers.8.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.8.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.8.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.8.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
"model.layers.8.self_attn.k_norm.weight": "model-00001-of-00004.safetensors",
"model.layers.8.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.8.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.8.self_attn.q_norm.weight": "model-00001-of-00004.safetensors",
"model.layers.8.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.8.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.9.input_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.9.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.9.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.9.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
"model.layers.9.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
"model.layers.9.self_attn.k_norm.weight": "model-00001-of-00004.safetensors",
"model.layers.9.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.9.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.9.self_attn.q_norm.weight": "model-00001-of-00004.safetensors",
"model.layers.9.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
"model.layers.9.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
"model.norm.weight": "model-00004-of-00004.safetensors"
}
}

31
special_tokens_map.json Normal file
View File

@@ -0,0 +1,31 @@
{
"additional_special_tokens": [
"<|im_start|>",
"<|im_end|>",
"<|object_ref_start|>",
"<|object_ref_end|>",
"<|box_start|>",
"<|box_end|>",
"<|quad_start|>",
"<|quad_end|>",
"<|vision_start|>",
"<|vision_end|>",
"<|vision_pad|>",
"<|image_pad|>",
"<|video_pad|>"
],
"eos_token": {
"content": "<|im_end|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false
},
"pad_token": {
"content": "<|endoftext|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false
}
}

BIN
tokenizer.json (Stored with Git LFS) Normal file

Binary file not shown.

240
tokenizer_config.json Normal file
View File

@@ -0,0 +1,240 @@
{
"add_bos_token": false,
"add_prefix_space": false,
"added_tokens_decoder": {
"151643": {
"content": "<|endoftext|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
},
"151644": {
"content": "<|im_start|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
},
"151645": {
"content": "<|im_end|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
},
"151646": {
"content": "<|object_ref_start|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
},
"151647": {
"content": "<|object_ref_end|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
},
"151648": {
"content": "<|box_start|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
},
"151649": {
"content": "<|box_end|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
},
"151650": {
"content": "<|quad_start|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
},
"151651": {
"content": "<|quad_end|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
},
"151652": {
"content": "<|vision_start|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
},
"151653": {
"content": "<|vision_end|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
},
"151654": {
"content": "<|vision_pad|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
},
"151655": {
"content": "<|image_pad|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
},
"151656": {
"content": "<|video_pad|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": true
},
"151657": {
"content": "<tool_call>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": false
},
"151658": {
"content": "</tool_call>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": false
},
"151659": {
"content": "<|fim_prefix|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": false
},
"151660": {
"content": "<|fim_middle|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": false
},
"151661": {
"content": "<|fim_suffix|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": false
},
"151662": {
"content": "<|fim_pad|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": false
},
"151663": {
"content": "<|repo_name|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": false
},
"151664": {
"content": "<|file_sep|>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": false
},
"151665": {
"content": "<tool_response>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": false
},
"151666": {
"content": "</tool_response>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": false
},
"151667": {
"content": "<think>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": false
},
"151668": {
"content": "</think>",
"lstrip": false,
"normalized": false,
"rstrip": false,
"single_word": false,
"special": false
}
},
"additional_special_tokens": [
"<|im_start|>",
"<|im_end|>",
"<|object_ref_start|>",
"<|object_ref_end|>",
"<|box_start|>",
"<|box_end|>",
"<|quad_start|>",
"<|quad_end|>",
"<|vision_start|>",
"<|vision_end|>",
"<|vision_pad|>",
"<|image_pad|>",
"<|video_pad|>"
],
"bos_token": null,
"clean_up_tokenization_spaces": false,
"eos_token": "<|im_end|>",
"errors": "replace",
"extra_special_tokens": {},
"model_max_length": 131072,
"pad_token": "<|endoftext|>",
"padding_side": "right",
"split_special_tokens": false,
"tokenizer_class": "Qwen2Tokenizer",
"unk_token": null
}

8
train_results.json Normal file
View File

@@ -0,0 +1,8 @@
{
"epoch": 3.0,
"total_flos": 3.3860293178247086e+18,
"train_loss": 0.5216725878735893,
"train_runtime": 16633.6833,
"train_samples_per_second": 0.449,
"train_steps_per_second": 0.007
}

118
trainer_log.jsonl Normal file
View File

@@ -0,0 +1,118 @@
{"current_steps": 1, "total_steps": 117, "loss": 0.7626, "lr": 0.0, "epoch": 0.02564102564102564, "percentage": 0.85, "elapsed_time": "0:02:51", "remaining_time": "5:31:45"}
{"current_steps": 2, "total_steps": 117, "loss": 0.7252, "lr": 8.333333333333333e-08, "epoch": 0.05128205128205128, "percentage": 1.71, "elapsed_time": "0:04:52", "remaining_time": "4:40:24"}
{"current_steps": 3, "total_steps": 117, "loss": 0.6479, "lr": 1.6666666666666665e-07, "epoch": 0.07692307692307693, "percentage": 2.56, "elapsed_time": "0:07:37", "remaining_time": "4:49:40"}
{"current_steps": 4, "total_steps": 117, "loss": 0.7078, "lr": 2.5e-07, "epoch": 0.10256410256410256, "percentage": 3.42, "elapsed_time": "0:09:54", "remaining_time": "4:40:01"}
{"current_steps": 5, "total_steps": 117, "loss": 0.4819, "lr": 3.333333333333333e-07, "epoch": 0.1282051282051282, "percentage": 4.27, "elapsed_time": "0:12:03", "remaining_time": "4:30:12"}
{"current_steps": 6, "total_steps": 117, "loss": 0.6797, "lr": 4.1666666666666667e-07, "epoch": 0.15384615384615385, "percentage": 5.13, "elapsed_time": "0:14:16", "remaining_time": "4:23:58"}
{"current_steps": 7, "total_steps": 117, "loss": 0.6077, "lr": 5e-07, "epoch": 0.1794871794871795, "percentage": 5.98, "elapsed_time": "0:16:11", "remaining_time": "4:14:24"}
{"current_steps": 8, "total_steps": 117, "loss": 0.7324, "lr": 5.833333333333334e-07, "epoch": 0.20512820512820512, "percentage": 6.84, "elapsed_time": "0:18:48", "remaining_time": "4:16:12"}
{"current_steps": 9, "total_steps": 117, "loss": 0.7341, "lr": 6.666666666666666e-07, "epoch": 0.23076923076923078, "percentage": 7.69, "elapsed_time": "0:21:33", "remaining_time": "4:18:36"}
{"current_steps": 10, "total_steps": 117, "loss": 0.6812, "lr": 7.5e-07, "epoch": 0.2564102564102564, "percentage": 8.55, "elapsed_time": "0:23:51", "remaining_time": "4:15:15"}
{"current_steps": 11, "total_steps": 117, "loss": 0.7673, "lr": 8.333333333333333e-07, "epoch": 0.28205128205128205, "percentage": 9.4, "elapsed_time": "0:26:24", "remaining_time": "4:14:29"}
{"current_steps": 12, "total_steps": 117, "loss": 0.7046, "lr": 9.166666666666665e-07, "epoch": 0.3076923076923077, "percentage": 10.26, "elapsed_time": "0:28:41", "remaining_time": "4:11:03"}
{"current_steps": 13, "total_steps": 117, "loss": 0.7147, "lr": 1e-06, "epoch": 0.3333333333333333, "percentage": 11.11, "elapsed_time": "0:31:37", "remaining_time": "4:12:59"}
{"current_steps": 14, "total_steps": 117, "loss": 0.7204, "lr": 9.997762161417517e-07, "epoch": 0.358974358974359, "percentage": 11.97, "elapsed_time": "0:34:08", "remaining_time": "4:11:11"}
{"current_steps": 15, "total_steps": 117, "loss": 0.6313, "lr": 9.991050648838675e-07, "epoch": 0.38461538461538464, "percentage": 12.82, "elapsed_time": "0:36:19", "remaining_time": "4:06:59"}
{"current_steps": 16, "total_steps": 117, "loss": 0.74, "lr": 9.979871469976195e-07, "epoch": 0.41025641025641024, "percentage": 13.68, "elapsed_time": "0:38:09", "remaining_time": "4:00:52"}
{"current_steps": 17, "total_steps": 117, "loss": 0.5999, "lr": 9.964234631709185e-07, "epoch": 0.4358974358974359, "percentage": 14.53, "elapsed_time": "0:40:13", "remaining_time": "3:56:39"}
{"current_steps": 18, "total_steps": 117, "loss": 0.5882, "lr": 9.944154131125642e-07, "epoch": 0.46153846153846156, "percentage": 15.38, "elapsed_time": "0:43:01", "remaining_time": "3:56:38"}
{"current_steps": 19, "total_steps": 117, "loss": 0.7462, "lr": 9.919647942993147e-07, "epoch": 0.48717948717948717, "percentage": 16.24, "elapsed_time": "0:44:55", "remaining_time": "3:51:41"}
{"current_steps": 20, "total_steps": 117, "loss": 0.6599, "lr": 9.890738003669027e-07, "epoch": 0.5128205128205128, "percentage": 17.09, "elapsed_time": "0:47:50", "remaining_time": "3:52:02"}
{"current_steps": 21, "total_steps": 117, "loss": 0.6324, "lr": 9.857450191464337e-07, "epoch": 0.5384615384615384, "percentage": 17.95, "elapsed_time": "0:50:09", "remaining_time": "3:49:16"}
{"current_steps": 22, "total_steps": 117, "loss": 0.5479, "lr": 9.819814303479267e-07, "epoch": 0.5641025641025641, "percentage": 18.8, "elapsed_time": "0:52:09", "remaining_time": "3:45:14"}
{"current_steps": 23, "total_steps": 117, "loss": 0.6751, "lr": 9.777864028930705e-07, "epoch": 0.5897435897435898, "percentage": 19.66, "elapsed_time": "0:53:33", "remaining_time": "3:38:54"}
{"current_steps": 24, "total_steps": 117, "loss": 0.5546, "lr": 9.73163691899582e-07, "epoch": 0.6153846153846154, "percentage": 20.51, "elapsed_time": "0:55:30", "remaining_time": "3:35:06"}
{"current_steps": 25, "total_steps": 117, "loss": 0.6147, "lr": 9.681174353198686e-07, "epoch": 0.6410256410256411, "percentage": 21.37, "elapsed_time": "0:58:02", "remaining_time": "3:33:34"}
{"current_steps": 26, "total_steps": 117, "loss": 0.5441, "lr": 9.626521502369983e-07, "epoch": 0.6666666666666666, "percentage": 22.22, "elapsed_time": "1:01:05", "remaining_time": "3:33:48"}
{"current_steps": 27, "total_steps": 117, "loss": 0.4269, "lr": 9.567727288213004e-07, "epoch": 0.6923076923076923, "percentage": 23.08, "elapsed_time": "1:03:42", "remaining_time": "3:32:21"}
{"current_steps": 28, "total_steps": 117, "loss": 0.4746, "lr": 9.504844339512094e-07, "epoch": 0.717948717948718, "percentage": 23.93, "elapsed_time": "1:05:25", "remaining_time": "3:27:57"}
{"current_steps": 29, "total_steps": 117, "loss": 0.5064, "lr": 9.43792894502277e-07, "epoch": 0.7435897435897436, "percentage": 24.79, "elapsed_time": "1:07:46", "remaining_time": "3:25:40"}
{"current_steps": 30, "total_steps": 117, "loss": 0.6522, "lr": 9.367041003085648e-07, "epoch": 0.7692307692307693, "percentage": 25.64, "elapsed_time": "1:10:02", "remaining_time": "3:23:06"}
{"current_steps": 31, "total_steps": 117, "loss": 0.5389, "lr": 9.29224396800933e-07, "epoch": 0.7948717948717948, "percentage": 26.5, "elapsed_time": "1:13:04", "remaining_time": "3:22:44"}
{"current_steps": 32, "total_steps": 117, "loss": 0.5355, "lr": 9.213604793270196e-07, "epoch": 0.8205128205128205, "percentage": 27.35, "elapsed_time": "1:15:49", "remaining_time": "3:21:24"}
{"current_steps": 33, "total_steps": 117, "loss": 0.4962, "lr": 9.131193871579974e-07, "epoch": 0.8461538461538461, "percentage": 28.21, "elapsed_time": "1:18:09", "remaining_time": "3:18:56"}
{"current_steps": 34, "total_steps": 117, "loss": 0.5688, "lr": 9.045084971874737e-07, "epoch": 0.8717948717948718, "percentage": 29.06, "elapsed_time": "1:19:34", "remaining_time": "3:14:15"}
{"current_steps": 35, "total_steps": 117, "loss": 0.5549, "lr": 8.955355173281707e-07, "epoch": 0.8974358974358975, "percentage": 29.91, "elapsed_time": "1:22:11", "remaining_time": "3:12:34"}
{"current_steps": 36, "total_steps": 117, "loss": 0.5553, "lr": 8.862084796122997e-07, "epoch": 0.9230769230769231, "percentage": 30.77, "elapsed_time": "1:23:58", "remaining_time": "3:08:57"}
{"current_steps": 37, "total_steps": 117, "loss": 0.5482, "lr": 8.765357330018055e-07, "epoch": 0.9487179487179487, "percentage": 31.62, "elapsed_time": "1:26:25", "remaining_time": "3:06:51"}
{"current_steps": 38, "total_steps": 117, "loss": 0.4679, "lr": 8.66525935914913e-07, "epoch": 0.9743589743589743, "percentage": 32.48, "elapsed_time": "1:29:15", "remaining_time": "3:05:33"}
{"current_steps": 39, "total_steps": 117, "loss": 0.3698, "lr": 8.561880484756724e-07, "epoch": 1.0, "percentage": 33.33, "elapsed_time": "1:31:43", "remaining_time": "3:03:27"}
{"current_steps": 40, "total_steps": 117, "loss": 0.5947, "lr": 8.455313244934324e-07, "epoch": 1.0256410256410255, "percentage": 34.19, "elapsed_time": "1:34:49", "remaining_time": "3:02:31"}
{"current_steps": 41, "total_steps": 117, "loss": 0.4995, "lr": 8.34565303179429e-07, "epoch": 1.0512820512820513, "percentage": 35.04, "elapsed_time": "1:37:04", "remaining_time": "2:59:56"}
{"current_steps": 42, "total_steps": 117, "loss": 0.5036, "lr": 8.232998006078997e-07, "epoch": 1.0769230769230769, "percentage": 35.9, "elapsed_time": "1:39:41", "remaining_time": "2:58:00"}
{"current_steps": 43, "total_steps": 117, "loss": 0.3792, "lr": 8.117449009293668e-07, "epoch": 1.1025641025641026, "percentage": 36.75, "elapsed_time": "1:42:27", "remaining_time": "2:56:19"}
{"current_steps": 44, "total_steps": 117, "loss": 0.516, "lr": 7.999109473439569e-07, "epoch": 1.1282051282051282, "percentage": 37.61, "elapsed_time": "1:44:39", "remaining_time": "2:53:38"}
{"current_steps": 45, "total_steps": 117, "loss": 0.5099, "lr": 7.878085328428368e-07, "epoch": 1.1538461538461537, "percentage": 38.46, "elapsed_time": "1:46:38", "remaining_time": "2:50:38"}
{"current_steps": 46, "total_steps": 117, "loss": 0.4209, "lr": 7.754484907260512e-07, "epoch": 1.1794871794871795, "percentage": 39.32, "elapsed_time": "1:49:07", "remaining_time": "2:48:25"}
{"current_steps": 47, "total_steps": 117, "loss": 0.4995, "lr": 7.628418849052523e-07, "epoch": 1.205128205128205, "percentage": 40.17, "elapsed_time": "1:51:38", "remaining_time": "2:46:15"}
{"current_steps": 48, "total_steps": 117, "loss": 0.5338, "lr": 7.5e-07, "epoch": 1.2307692307692308, "percentage": 41.03, "elapsed_time": "1:53:37", "remaining_time": "2:43:20"}
{"current_steps": 49, "total_steps": 117, "loss": 0.4971, "lr": 7.369343312364993e-07, "epoch": 1.2564102564102564, "percentage": 41.88, "elapsed_time": "1:55:28", "remaining_time": "2:40:15"}
{"current_steps": 50, "total_steps": 117, "loss": 0.5607, "lr": 7.236565741578162e-07, "epoch": 1.282051282051282, "percentage": 42.74, "elapsed_time": "1:57:15", "remaining_time": "2:37:08"}
{"current_steps": 51, "total_steps": 117, "loss": 0.3963, "lr": 7.101786141547828e-07, "epoch": 1.3076923076923077, "percentage": 43.59, "elapsed_time": "1:59:47", "remaining_time": "2:35:02"}
{"current_steps": 52, "total_steps": 117, "loss": 0.4136, "lr": 6.965125158269618e-07, "epoch": 1.3333333333333333, "percentage": 44.44, "elapsed_time": "2:01:42", "remaining_time": "2:32:07"}
{"current_steps": 53, "total_steps": 117, "loss": 0.4876, "lr": 6.826705121831976e-07, "epoch": 1.358974358974359, "percentage": 45.3, "elapsed_time": "2:04:19", "remaining_time": "2:30:07"}
{"current_steps": 54, "total_steps": 117, "loss": 0.5899, "lr": 6.68664993691415e-07, "epoch": 1.3846153846153846, "percentage": 46.15, "elapsed_time": "2:06:30", "remaining_time": "2:27:35"}
{"current_steps": 55, "total_steps": 117, "loss": 0.5163, "lr": 6.545084971874736e-07, "epoch": 1.4102564102564101, "percentage": 47.01, "elapsed_time": "2:08:52", "remaining_time": "2:25:16"}
{"current_steps": 56, "total_steps": 117, "loss": 0.4375, "lr": 6.402136946530014e-07, "epoch": 1.435897435897436, "percentage": 47.86, "elapsed_time": "2:11:12", "remaining_time": "2:22:55"}
{"current_steps": 57, "total_steps": 117, "loss": 0.4492, "lr": 6.257933818722542e-07, "epoch": 1.4615384615384617, "percentage": 48.72, "elapsed_time": "2:13:33", "remaining_time": "2:20:35"}
{"current_steps": 58, "total_steps": 117, "loss": 0.4733, "lr": 6.112604669781572e-07, "epoch": 1.4871794871794872, "percentage": 49.57, "elapsed_time": "2:16:01", "remaining_time": "2:18:21"}
{"current_steps": 59, "total_steps": 117, "loss": 0.4164, "lr": 5.966279588977766e-07, "epoch": 1.5128205128205128, "percentage": 50.43, "elapsed_time": "2:18:36", "remaining_time": "2:16:15"}
{"current_steps": 60, "total_steps": 117, "loss": 0.477, "lr": 5.819089557075688e-07, "epoch": 1.5384615384615383, "percentage": 51.28, "elapsed_time": "2:20:44", "remaining_time": "2:13:42"}
{"current_steps": 61, "total_steps": 117, "loss": 0.4343, "lr": 5.671166329088277e-07, "epoch": 1.564102564102564, "percentage": 52.14, "elapsed_time": "2:23:39", "remaining_time": "2:11:52"}
{"current_steps": 62, "total_steps": 117, "loss": 0.5995, "lr": 5.522642316338268e-07, "epoch": 1.5897435897435899, "percentage": 52.99, "elapsed_time": "2:26:04", "remaining_time": "2:09:34"}
{"current_steps": 63, "total_steps": 117, "loss": 0.5178, "lr": 5.373650467932121e-07, "epoch": 1.6153846153846154, "percentage": 53.85, "elapsed_time": "2:28:32", "remaining_time": "2:07:19"}
{"current_steps": 64, "total_steps": 117, "loss": 0.6536, "lr": 5.224324151752575e-07, "epoch": 1.641025641025641, "percentage": 54.7, "elapsed_time": "2:31:35", "remaining_time": "2:05:31"}
{"current_steps": 65, "total_steps": 117, "loss": 0.3648, "lr": 5.074797035076318e-07, "epoch": 1.6666666666666665, "percentage": 55.56, "elapsed_time": "2:34:28", "remaining_time": "2:03:34"}
{"current_steps": 66, "total_steps": 117, "loss": 0.4617, "lr": 4.925202964923683e-07, "epoch": 1.6923076923076923, "percentage": 56.41, "elapsed_time": "2:36:16", "remaining_time": "2:00:45"}
{"current_steps": 67, "total_steps": 117, "loss": 0.5203, "lr": 4.775675848247427e-07, "epoch": 1.717948717948718, "percentage": 57.26, "elapsed_time": "2:38:26", "remaining_time": "1:58:14"}
{"current_steps": 68, "total_steps": 117, "loss": 0.4732, "lr": 4.626349532067879e-07, "epoch": 1.7435897435897436, "percentage": 58.12, "elapsed_time": "2:40:42", "remaining_time": "1:55:48"}
{"current_steps": 69, "total_steps": 117, "loss": 0.3996, "lr": 4.477357683661733e-07, "epoch": 1.7692307692307692, "percentage": 58.97, "elapsed_time": "2:43:45", "remaining_time": "1:53:55"}
{"current_steps": 70, "total_steps": 117, "loss": 0.4761, "lr": 4.328833670911724e-07, "epoch": 1.7948717948717947, "percentage": 59.83, "elapsed_time": "2:45:39", "remaining_time": "1:51:13"}
{"current_steps": 71, "total_steps": 117, "loss": 0.4358, "lr": 4.180910442924311e-07, "epoch": 1.8205128205128205, "percentage": 60.68, "elapsed_time": "2:48:05", "remaining_time": "1:48:54"}
{"current_steps": 72, "total_steps": 117, "loss": 0.4984, "lr": 4.0337204110222347e-07, "epoch": 1.8461538461538463, "percentage": 61.54, "elapsed_time": "2:50:57", "remaining_time": "1:46:50"}
{"current_steps": 73, "total_steps": 117, "loss": 0.4777, "lr": 3.8873953302184283e-07, "epoch": 1.8717948717948718, "percentage": 62.39, "elapsed_time": "2:53:27", "remaining_time": "1:44:32"}
{"current_steps": 74, "total_steps": 117, "loss": 0.47, "lr": 3.742066181277457e-07, "epoch": 1.8974358974358974, "percentage": 63.25, "elapsed_time": "2:55:07", "remaining_time": "1:41:45"}
{"current_steps": 75, "total_steps": 117, "loss": 0.4638, "lr": 3.5978630534699865e-07, "epoch": 1.9230769230769231, "percentage": 64.1, "elapsed_time": "2:57:32", "remaining_time": "1:39:25"}
{"current_steps": 76, "total_steps": 117, "loss": 0.3967, "lr": 3.454915028125263e-07, "epoch": 1.9487179487179487, "percentage": 64.96, "elapsed_time": "3:00:02", "remaining_time": "1:37:07"}
{"current_steps": 77, "total_steps": 117, "loss": 0.5487, "lr": 3.31335006308585e-07, "epoch": 1.9743589743589745, "percentage": 65.81, "elapsed_time": "3:01:29", "remaining_time": "1:34:17"}
{"current_steps": 78, "total_steps": 117, "loss": 0.4027, "lr": 3.173294878168025e-07, "epoch": 2.0, "percentage": 66.67, "elapsed_time": "3:03:56", "remaining_time": "1:31:58"}
{"current_steps": 79, "total_steps": 117, "loss": 0.5068, "lr": 3.034874841730382e-07, "epoch": 2.0256410256410255, "percentage": 67.52, "elapsed_time": "3:06:51", "remaining_time": "1:29:53"}
{"current_steps": 80, "total_steps": 117, "loss": 0.4841, "lr": 2.898213858452173e-07, "epoch": 2.051282051282051, "percentage": 68.38, "elapsed_time": "3:09:00", "remaining_time": "1:27:25"}
{"current_steps": 81, "total_steps": 117, "loss": 0.4362, "lr": 2.763434258421836e-07, "epoch": 2.076923076923077, "percentage": 69.23, "elapsed_time": "3:11:12", "remaining_time": "1:24:58"}
{"current_steps": 82, "total_steps": 117, "loss": 0.5247, "lr": 2.6306566876350067e-07, "epoch": 2.1025641025641026, "percentage": 70.09, "elapsed_time": "3:13:13", "remaining_time": "1:22:28"}
{"current_steps": 83, "total_steps": 117, "loss": 0.4988, "lr": 2.500000000000001e-07, "epoch": 2.128205128205128, "percentage": 70.94, "elapsed_time": "3:15:54", "remaining_time": "1:20:14"}
{"current_steps": 84, "total_steps": 117, "loss": 0.4818, "lr": 2.371581150947476e-07, "epoch": 2.1538461538461537, "percentage": 71.79, "elapsed_time": "3:18:20", "remaining_time": "1:17:55"}
{"current_steps": 85, "total_steps": 117, "loss": 0.4962, "lr": 2.2455150927394878e-07, "epoch": 2.1794871794871793, "percentage": 72.65, "elapsed_time": "3:20:27", "remaining_time": "1:15:28"}
{"current_steps": 86, "total_steps": 117, "loss": 0.4044, "lr": 2.121914671571633e-07, "epoch": 2.2051282051282053, "percentage": 73.5, "elapsed_time": "3:22:55", "remaining_time": "1:13:08"}
{"current_steps": 87, "total_steps": 117, "loss": 0.4197, "lr": 2.0008905265604315e-07, "epoch": 2.230769230769231, "percentage": 74.36, "elapsed_time": "3:25:12", "remaining_time": "1:10:45"}
{"current_steps": 88, "total_steps": 117, "loss": 0.4524, "lr": 1.8825509907063326e-07, "epoch": 2.2564102564102564, "percentage": 75.21, "elapsed_time": "3:27:09", "remaining_time": "1:08:16"}
{"current_steps": 89, "total_steps": 117, "loss": 0.5084, "lr": 1.7670019939210023e-07, "epoch": 2.282051282051282, "percentage": 76.07, "elapsed_time": "3:29:26", "remaining_time": "1:05:53"}
{"current_steps": 90, "total_steps": 117, "loss": 0.4534, "lr": 1.6543469682057104e-07, "epoch": 2.3076923076923075, "percentage": 76.92, "elapsed_time": "3:32:03", "remaining_time": "1:03:36"}
{"current_steps": 91, "total_steps": 117, "loss": 0.5118, "lr": 1.5446867550656767e-07, "epoch": 2.3333333333333335, "percentage": 77.78, "elapsed_time": "3:34:30", "remaining_time": "1:01:17"}
{"current_steps": 92, "total_steps": 117, "loss": 0.5472, "lr": 1.4381195152432769e-07, "epoch": 2.358974358974359, "percentage": 78.63, "elapsed_time": "3:36:50", "remaining_time": "0:58:55"}
{"current_steps": 93, "total_steps": 117, "loss": 0.3968, "lr": 1.3347406408508694e-07, "epoch": 2.3846153846153846, "percentage": 79.49, "elapsed_time": "3:39:17", "remaining_time": "0:56:35"}
{"current_steps": 94, "total_steps": 117, "loss": 0.4768, "lr": 1.2346426699819456e-07, "epoch": 2.41025641025641, "percentage": 80.34, "elapsed_time": "3:41:33", "remaining_time": "0:54:12"}
{"current_steps": 95, "total_steps": 117, "loss": 0.445, "lr": 1.1379152038770029e-07, "epoch": 2.435897435897436, "percentage": 81.2, "elapsed_time": "3:43:14", "remaining_time": "0:51:41"}
{"current_steps": 96, "total_steps": 117, "loss": 0.3917, "lr": 1.044644826718295e-07, "epoch": 2.4615384615384617, "percentage": 82.05, "elapsed_time": "3:45:39", "remaining_time": "0:49:21"}
{"current_steps": 97, "total_steps": 117, "loss": 0.5053, "lr": 9.549150281252632e-08, "epoch": 2.4871794871794872, "percentage": 82.91, "elapsed_time": "3:48:09", "remaining_time": "0:47:02"}
{"current_steps": 98, "total_steps": 117, "loss": 0.4785, "lr": 8.688061284200265e-08, "epoch": 2.5128205128205128, "percentage": 83.76, "elapsed_time": "3:50:35", "remaining_time": "0:44:42"}
{"current_steps": 99, "total_steps": 117, "loss": 0.4962, "lr": 7.863952067298041e-08, "epoch": 2.5384615384615383, "percentage": 84.62, "elapsed_time": "3:52:51", "remaining_time": "0:42:20"}
{"current_steps": 100, "total_steps": 117, "loss": 0.508, "lr": 7.077560319906694e-08, "epoch": 2.564102564102564, "percentage": 85.47, "elapsed_time": "3:54:58", "remaining_time": "0:39:56"}
{"current_steps": 101, "total_steps": 117, "loss": 0.4198, "lr": 6.329589969143517e-08, "epoch": 2.58974358974359, "percentage": 86.32, "elapsed_time": "3:58:01", "remaining_time": "0:37:42"}
{"current_steps": 102, "total_steps": 117, "loss": 0.4336, "lr": 5.620710549772295e-08, "epoch": 2.6153846153846154, "percentage": 87.18, "elapsed_time": "4:00:26", "remaining_time": "0:35:21"}
{"current_steps": 103, "total_steps": 117, "loss": 0.4482, "lr": 4.951556604879048e-08, "epoch": 2.641025641025641, "percentage": 88.03, "elapsed_time": "4:02:41", "remaining_time": "0:32:59"}
{"current_steps": 104, "total_steps": 117, "loss": 0.4978, "lr": 4.322727117869951e-08, "epoch": 2.6666666666666665, "percentage": 88.89, "elapsed_time": "4:05:22", "remaining_time": "0:30:40"}
{"current_steps": 105, "total_steps": 117, "loss": 0.5465, "lr": 3.734784976300165e-08, "epoch": 2.6923076923076925, "percentage": 89.74, "elapsed_time": "4:08:01", "remaining_time": "0:28:20"}
{"current_steps": 106, "total_steps": 117, "loss": 0.4831, "lr": 3.188256468013139e-08, "epoch": 2.717948717948718, "percentage": 90.6, "elapsed_time": "4:10:11", "remaining_time": "0:25:57"}
{"current_steps": 107, "total_steps": 117, "loss": 0.4515, "lr": 2.6836308100417872e-08, "epoch": 2.7435897435897436, "percentage": 91.45, "elapsed_time": "4:13:01", "remaining_time": "0:23:38"}
{"current_steps": 108, "total_steps": 117, "loss": 0.4528, "lr": 2.2213597106929605e-08, "epoch": 2.769230769230769, "percentage": 92.31, "elapsed_time": "4:15:24", "remaining_time": "0:21:17"}
{"current_steps": 109, "total_steps": 117, "loss": 0.492, "lr": 1.8018569652073378e-08, "epoch": 2.7948717948717947, "percentage": 93.16, "elapsed_time": "4:17:54", "remaining_time": "0:18:55"}
{"current_steps": 110, "total_steps": 117, "loss": 0.3204, "lr": 1.4254980853566246e-08, "epoch": 2.8205128205128203, "percentage": 94.02, "elapsed_time": "4:20:38", "remaining_time": "0:16:35"}
{"current_steps": 111, "total_steps": 117, "loss": 0.5362, "lr": 1.0926199633097154e-08, "epoch": 2.8461538461538463, "percentage": 94.87, "elapsed_time": "4:22:46", "remaining_time": "0:14:12"}
{"current_steps": 112, "total_steps": 117, "loss": 0.5113, "lr": 8.035205700685165e-09, "epoch": 2.871794871794872, "percentage": 95.73, "elapsed_time": "4:25:03", "remaining_time": "0:11:49"}
{"current_steps": 113, "total_steps": 117, "loss": 0.4808, "lr": 5.5845868874357385e-09, "epoch": 2.8974358974358974, "percentage": 96.58, "elapsed_time": "4:27:29", "remaining_time": "0:09:28"}
{"current_steps": 114, "total_steps": 117, "loss": 0.414, "lr": 3.5765368290813223e-09, "epoch": 2.9230769230769234, "percentage": 97.44, "elapsed_time": "4:30:06", "remaining_time": "0:07:06"}
{"current_steps": 115, "total_steps": 117, "loss": 0.3637, "lr": 2.0128530023804656e-09, "epoch": 2.948717948717949, "percentage": 98.29, "elapsed_time": "4:32:32", "remaining_time": "0:04:44"}
{"current_steps": 116, "total_steps": 117, "loss": 0.4558, "lr": 8.949351161324225e-10, "epoch": 2.9743589743589745, "percentage": 99.15, "elapsed_time": "4:35:10", "remaining_time": "0:02:22"}
{"current_steps": 117, "total_steps": 117, "loss": 0.6395, "lr": 2.2378385824833866e-10, "epoch": 3.0, "percentage": 100.0, "elapsed_time": "4:36:36", "remaining_time": "0:00:00"}
{"current_steps": 117, "total_steps": 117, "epoch": 3.0, "percentage": 100.0, "elapsed_time": "4:37:13", "remaining_time": "0:00:00"}

862
trainer_state.json Normal file
View File

@@ -0,0 +1,862 @@
{
"best_global_step": null,
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 3.0,
"eval_steps": 500,
"global_step": 117,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.02564102564102564,
"grad_norm": 6.369640827178955,
"learning_rate": 0.0,
"loss": 0.7626,
"step": 1
},
{
"epoch": 0.05128205128205128,
"grad_norm": 6.251739025115967,
"learning_rate": 8.333333333333333e-08,
"loss": 0.7252,
"step": 2
},
{
"epoch": 0.07692307692307693,
"grad_norm": 6.172633171081543,
"learning_rate": 1.6666666666666665e-07,
"loss": 0.6479,
"step": 3
},
{
"epoch": 0.10256410256410256,
"grad_norm": 5.884990692138672,
"learning_rate": 2.5e-07,
"loss": 0.7078,
"step": 4
},
{
"epoch": 0.1282051282051282,
"grad_norm": 5.081572532653809,
"learning_rate": 3.333333333333333e-07,
"loss": 0.4819,
"step": 5
},
{
"epoch": 0.15384615384615385,
"grad_norm": 6.8136091232299805,
"learning_rate": 4.1666666666666667e-07,
"loss": 0.6797,
"step": 6
},
{
"epoch": 0.1794871794871795,
"grad_norm": 6.076236724853516,
"learning_rate": 5e-07,
"loss": 0.6077,
"step": 7
},
{
"epoch": 0.20512820512820512,
"grad_norm": 6.338658809661865,
"learning_rate": 5.833333333333334e-07,
"loss": 0.7324,
"step": 8
},
{
"epoch": 0.23076923076923078,
"grad_norm": 6.2719292640686035,
"learning_rate": 6.666666666666666e-07,
"loss": 0.7341,
"step": 9
},
{
"epoch": 0.2564102564102564,
"grad_norm": 5.779564380645752,
"learning_rate": 7.5e-07,
"loss": 0.6812,
"step": 10
},
{
"epoch": 0.28205128205128205,
"grad_norm": 6.111364841461182,
"learning_rate": 8.333333333333333e-07,
"loss": 0.7673,
"step": 11
},
{
"epoch": 0.3076923076923077,
"grad_norm": 5.112427234649658,
"learning_rate": 9.166666666666665e-07,
"loss": 0.7046,
"step": 12
},
{
"epoch": 0.3333333333333333,
"grad_norm": 4.599231719970703,
"learning_rate": 1e-06,
"loss": 0.7147,
"step": 13
},
{
"epoch": 0.358974358974359,
"grad_norm": 4.481863021850586,
"learning_rate": 9.997762161417517e-07,
"loss": 0.7204,
"step": 14
},
{
"epoch": 0.38461538461538464,
"grad_norm": 4.414555549621582,
"learning_rate": 9.991050648838675e-07,
"loss": 0.6313,
"step": 15
},
{
"epoch": 0.41025641025641024,
"grad_norm": 3.1873695850372314,
"learning_rate": 9.979871469976195e-07,
"loss": 0.74,
"step": 16
},
{
"epoch": 0.4358974358974359,
"grad_norm": 2.990138530731201,
"learning_rate": 9.964234631709185e-07,
"loss": 0.5999,
"step": 17
},
{
"epoch": 0.46153846153846156,
"grad_norm": 2.494980573654175,
"learning_rate": 9.944154131125642e-07,
"loss": 0.5882,
"step": 18
},
{
"epoch": 0.48717948717948717,
"grad_norm": 3.04402232170105,
"learning_rate": 9.919647942993147e-07,
"loss": 0.7462,
"step": 19
},
{
"epoch": 0.5128205128205128,
"grad_norm": 2.4796009063720703,
"learning_rate": 9.890738003669027e-07,
"loss": 0.6599,
"step": 20
},
{
"epoch": 0.5384615384615384,
"grad_norm": 2.6610946655273438,
"learning_rate": 9.857450191464337e-07,
"loss": 0.6324,
"step": 21
},
{
"epoch": 0.5641025641025641,
"grad_norm": 2.2293684482574463,
"learning_rate": 9.819814303479267e-07,
"loss": 0.5479,
"step": 22
},
{
"epoch": 0.5897435897435898,
"grad_norm": 2.254640817642212,
"learning_rate": 9.777864028930705e-07,
"loss": 0.6751,
"step": 23
},
{
"epoch": 0.6153846153846154,
"grad_norm": 1.634244680404663,
"learning_rate": 9.73163691899582e-07,
"loss": 0.5546,
"step": 24
},
{
"epoch": 0.6410256410256411,
"grad_norm": 1.6159415245056152,
"learning_rate": 9.681174353198686e-07,
"loss": 0.6147,
"step": 25
},
{
"epoch": 0.6666666666666666,
"grad_norm": 1.4406708478927612,
"learning_rate": 9.626521502369983e-07,
"loss": 0.5441,
"step": 26
},
{
"epoch": 0.6923076923076923,
"grad_norm": 1.068778157234192,
"learning_rate": 9.567727288213004e-07,
"loss": 0.4269,
"step": 27
},
{
"epoch": 0.717948717948718,
"grad_norm": 1.2805856466293335,
"learning_rate": 9.504844339512094e-07,
"loss": 0.4746,
"step": 28
},
{
"epoch": 0.7435897435897436,
"grad_norm": 1.2707418203353882,
"learning_rate": 9.43792894502277e-07,
"loss": 0.5064,
"step": 29
},
{
"epoch": 0.7692307692307693,
"grad_norm": 1.3852788209915161,
"learning_rate": 9.367041003085648e-07,
"loss": 0.6522,
"step": 30
},
{
"epoch": 0.7948717948717948,
"grad_norm": 1.1346880197525024,
"learning_rate": 9.29224396800933e-07,
"loss": 0.5389,
"step": 31
},
{
"epoch": 0.8205128205128205,
"grad_norm": 1.026390790939331,
"learning_rate": 9.213604793270196e-07,
"loss": 0.5355,
"step": 32
},
{
"epoch": 0.8461538461538461,
"grad_norm": 0.9689812064170837,
"learning_rate": 9.131193871579974e-07,
"loss": 0.4962,
"step": 33
},
{
"epoch": 0.8717948717948718,
"grad_norm": 0.9934576749801636,
"learning_rate": 9.045084971874737e-07,
"loss": 0.5688,
"step": 34
},
{
"epoch": 0.8974358974358975,
"grad_norm": 0.8792146444320679,
"learning_rate": 8.955355173281707e-07,
"loss": 0.5549,
"step": 35
},
{
"epoch": 0.9230769230769231,
"grad_norm": 0.8279433250427246,
"learning_rate": 8.862084796122997e-07,
"loss": 0.5553,
"step": 36
},
{
"epoch": 0.9487179487179487,
"grad_norm": 0.7716125249862671,
"learning_rate": 8.765357330018055e-07,
"loss": 0.5482,
"step": 37
},
{
"epoch": 0.9743589743589743,
"grad_norm": 0.5753526091575623,
"learning_rate": 8.66525935914913e-07,
"loss": 0.4679,
"step": 38
},
{
"epoch": 1.0,
"grad_norm": 0.48458731174468994,
"learning_rate": 8.561880484756724e-07,
"loss": 0.3698,
"step": 39
},
{
"epoch": 1.0256410256410255,
"grad_norm": 0.6180218458175659,
"learning_rate": 8.455313244934324e-07,
"loss": 0.5947,
"step": 40
},
{
"epoch": 1.0512820512820513,
"grad_norm": 0.7078486680984497,
"learning_rate": 8.34565303179429e-07,
"loss": 0.4995,
"step": 41
},
{
"epoch": 1.0769230769230769,
"grad_norm": 0.5181483030319214,
"learning_rate": 8.232998006078997e-07,
"loss": 0.5036,
"step": 42
},
{
"epoch": 1.1025641025641026,
"grad_norm": 0.45219916105270386,
"learning_rate": 8.117449009293668e-07,
"loss": 0.3792,
"step": 43
},
{
"epoch": 1.1282051282051282,
"grad_norm": 0.4868799149990082,
"learning_rate": 7.999109473439569e-07,
"loss": 0.516,
"step": 44
},
{
"epoch": 1.1538461538461537,
"grad_norm": 0.5245233178138733,
"learning_rate": 7.878085328428368e-07,
"loss": 0.5099,
"step": 45
},
{
"epoch": 1.1794871794871795,
"grad_norm": 0.4330838620662689,
"learning_rate": 7.754484907260512e-07,
"loss": 0.4209,
"step": 46
},
{
"epoch": 1.205128205128205,
"grad_norm": 0.4760264456272125,
"learning_rate": 7.628418849052523e-07,
"loss": 0.4995,
"step": 47
},
{
"epoch": 1.2307692307692308,
"grad_norm": 0.49813446402549744,
"learning_rate": 7.5e-07,
"loss": 0.5338,
"step": 48
},
{
"epoch": 1.2564102564102564,
"grad_norm": 0.45343366265296936,
"learning_rate": 7.369343312364993e-07,
"loss": 0.4971,
"step": 49
},
{
"epoch": 1.282051282051282,
"grad_norm": 0.4793028235435486,
"learning_rate": 7.236565741578162e-07,
"loss": 0.5607,
"step": 50
},
{
"epoch": 1.3076923076923077,
"grad_norm": 0.3807591199874878,
"learning_rate": 7.101786141547828e-07,
"loss": 0.3963,
"step": 51
},
{
"epoch": 1.3333333333333333,
"grad_norm": 0.4242149591445923,
"learning_rate": 6.965125158269618e-07,
"loss": 0.4136,
"step": 52
},
{
"epoch": 1.358974358974359,
"grad_norm": 0.5695387125015259,
"learning_rate": 6.826705121831976e-07,
"loss": 0.4876,
"step": 53
},
{
"epoch": 1.3846153846153846,
"grad_norm": 0.48757168650627136,
"learning_rate": 6.68664993691415e-07,
"loss": 0.5899,
"step": 54
},
{
"epoch": 1.4102564102564101,
"grad_norm": 0.4394301474094391,
"learning_rate": 6.545084971874736e-07,
"loss": 0.5163,
"step": 55
},
{
"epoch": 1.435897435897436,
"grad_norm": 0.38720548152923584,
"learning_rate": 6.402136946530014e-07,
"loss": 0.4375,
"step": 56
},
{
"epoch": 1.4615384615384617,
"grad_norm": 0.3783684968948364,
"learning_rate": 6.257933818722542e-07,
"loss": 0.4492,
"step": 57
},
{
"epoch": 1.4871794871794872,
"grad_norm": 0.4112834632396698,
"learning_rate": 6.112604669781572e-07,
"loss": 0.4733,
"step": 58
},
{
"epoch": 1.5128205128205128,
"grad_norm": 0.4309942424297333,
"learning_rate": 5.966279588977766e-07,
"loss": 0.4164,
"step": 59
},
{
"epoch": 1.5384615384615383,
"grad_norm": 0.4247504472732544,
"learning_rate": 5.819089557075688e-07,
"loss": 0.477,
"step": 60
},
{
"epoch": 1.564102564102564,
"grad_norm": 0.38459497690200806,
"learning_rate": 5.671166329088277e-07,
"loss": 0.4343,
"step": 61
},
{
"epoch": 1.5897435897435899,
"grad_norm": 0.4865002930164337,
"learning_rate": 5.522642316338268e-07,
"loss": 0.5995,
"step": 62
},
{
"epoch": 1.6153846153846154,
"grad_norm": 0.3904992341995239,
"learning_rate": 5.373650467932121e-07,
"loss": 0.5178,
"step": 63
},
{
"epoch": 1.641025641025641,
"grad_norm": 0.4682792127132416,
"learning_rate": 5.224324151752575e-07,
"loss": 0.6536,
"step": 64
},
{
"epoch": 1.6666666666666665,
"grad_norm": 0.31421154737472534,
"learning_rate": 5.074797035076318e-07,
"loss": 0.3648,
"step": 65
},
{
"epoch": 1.6923076923076923,
"grad_norm": 0.39036867022514343,
"learning_rate": 4.925202964923683e-07,
"loss": 0.4617,
"step": 66
},
{
"epoch": 1.717948717948718,
"grad_norm": 0.42162591218948364,
"learning_rate": 4.775675848247427e-07,
"loss": 0.5203,
"step": 67
},
{
"epoch": 1.7435897435897436,
"grad_norm": 0.37007007002830505,
"learning_rate": 4.626349532067879e-07,
"loss": 0.4732,
"step": 68
},
{
"epoch": 1.7692307692307692,
"grad_norm": 0.3757108747959137,
"learning_rate": 4.477357683661733e-07,
"loss": 0.3996,
"step": 69
},
{
"epoch": 1.7948717948717947,
"grad_norm": 0.39338812232017517,
"learning_rate": 4.328833670911724e-07,
"loss": 0.4761,
"step": 70
},
{
"epoch": 1.8205128205128205,
"grad_norm": 0.3347456455230713,
"learning_rate": 4.180910442924311e-07,
"loss": 0.4358,
"step": 71
},
{
"epoch": 1.8461538461538463,
"grad_norm": 0.40436047315597534,
"learning_rate": 4.0337204110222347e-07,
"loss": 0.4984,
"step": 72
},
{
"epoch": 1.8717948717948718,
"grad_norm": 0.40149423480033875,
"learning_rate": 3.8873953302184283e-07,
"loss": 0.4777,
"step": 73
},
{
"epoch": 1.8974358974358974,
"grad_norm": 0.42441582679748535,
"learning_rate": 3.742066181277457e-07,
"loss": 0.47,
"step": 74
},
{
"epoch": 1.9230769230769231,
"grad_norm": 0.3688524663448334,
"learning_rate": 3.5978630534699865e-07,
"loss": 0.4638,
"step": 75
},
{
"epoch": 1.9487179487179487,
"grad_norm": 0.36299046874046326,
"learning_rate": 3.454915028125263e-07,
"loss": 0.3967,
"step": 76
},
{
"epoch": 1.9743589743589745,
"grad_norm": 0.46353092789649963,
"learning_rate": 3.31335006308585e-07,
"loss": 0.5487,
"step": 77
},
{
"epoch": 2.0,
"grad_norm": 0.32297879457473755,
"learning_rate": 3.173294878168025e-07,
"loss": 0.4027,
"step": 78
},
{
"epoch": 2.0256410256410255,
"grad_norm": 0.37267327308654785,
"learning_rate": 3.034874841730382e-07,
"loss": 0.5068,
"step": 79
},
{
"epoch": 2.051282051282051,
"grad_norm": 0.3740202486515045,
"learning_rate": 2.898213858452173e-07,
"loss": 0.4841,
"step": 80
},
{
"epoch": 2.076923076923077,
"grad_norm": 0.37761014699935913,
"learning_rate": 2.763434258421836e-07,
"loss": 0.4362,
"step": 81
},
{
"epoch": 2.1025641025641026,
"grad_norm": 0.3903439939022064,
"learning_rate": 2.6306566876350067e-07,
"loss": 0.5247,
"step": 82
},
{
"epoch": 2.128205128205128,
"grad_norm": 0.3678279221057892,
"learning_rate": 2.500000000000001e-07,
"loss": 0.4988,
"step": 83
},
{
"epoch": 2.1538461538461537,
"grad_norm": 0.3791114091873169,
"learning_rate": 2.371581150947476e-07,
"loss": 0.4818,
"step": 84
},
{
"epoch": 2.1794871794871793,
"grad_norm": 0.3990109860897064,
"learning_rate": 2.2455150927394878e-07,
"loss": 0.4962,
"step": 85
},
{
"epoch": 2.2051282051282053,
"grad_norm": 0.36887475848197937,
"learning_rate": 2.121914671571633e-07,
"loss": 0.4044,
"step": 86
},
{
"epoch": 2.230769230769231,
"grad_norm": 0.372435986995697,
"learning_rate": 2.0008905265604315e-07,
"loss": 0.4197,
"step": 87
},
{
"epoch": 2.2564102564102564,
"grad_norm": 0.35440200567245483,
"learning_rate": 1.8825509907063326e-07,
"loss": 0.4524,
"step": 88
},
{
"epoch": 2.282051282051282,
"grad_norm": 0.4030032157897949,
"learning_rate": 1.7670019939210023e-07,
"loss": 0.5084,
"step": 89
},
{
"epoch": 2.3076923076923075,
"grad_norm": 0.38129332661628723,
"learning_rate": 1.6543469682057104e-07,
"loss": 0.4534,
"step": 90
},
{
"epoch": 2.3333333333333335,
"grad_norm": 0.37370947003364563,
"learning_rate": 1.5446867550656767e-07,
"loss": 0.5118,
"step": 91
},
{
"epoch": 2.358974358974359,
"grad_norm": 0.4511595368385315,
"learning_rate": 1.4381195152432769e-07,
"loss": 0.5472,
"step": 92
},
{
"epoch": 2.3846153846153846,
"grad_norm": 0.38010352849960327,
"learning_rate": 1.3347406408508694e-07,
"loss": 0.3968,
"step": 93
},
{
"epoch": 2.41025641025641,
"grad_norm": 0.39101359248161316,
"learning_rate": 1.2346426699819456e-07,
"loss": 0.4768,
"step": 94
},
{
"epoch": 2.435897435897436,
"grad_norm": 0.3871517777442932,
"learning_rate": 1.1379152038770029e-07,
"loss": 0.445,
"step": 95
},
{
"epoch": 2.4615384615384617,
"grad_norm": 0.29811641573905945,
"learning_rate": 1.044644826718295e-07,
"loss": 0.3917,
"step": 96
},
{
"epoch": 2.4871794871794872,
"grad_norm": 0.3692609369754791,
"learning_rate": 9.549150281252632e-08,
"loss": 0.5053,
"step": 97
},
{
"epoch": 2.5128205128205128,
"grad_norm": 0.3917727470397949,
"learning_rate": 8.688061284200265e-08,
"loss": 0.4785,
"step": 98
},
{
"epoch": 2.5384615384615383,
"grad_norm": 0.34881332516670227,
"learning_rate": 7.863952067298041e-08,
"loss": 0.4962,
"step": 99
},
{
"epoch": 2.564102564102564,
"grad_norm": 0.34965211153030396,
"learning_rate": 7.077560319906694e-08,
"loss": 0.508,
"step": 100
},
{
"epoch": 2.58974358974359,
"grad_norm": 0.34777775406837463,
"learning_rate": 6.329589969143517e-08,
"loss": 0.4198,
"step": 101
},
{
"epoch": 2.6153846153846154,
"grad_norm": 0.34757086634635925,
"learning_rate": 5.620710549772295e-08,
"loss": 0.4336,
"step": 102
},
{
"epoch": 2.641025641025641,
"grad_norm": 0.36974871158599854,
"learning_rate": 4.951556604879048e-08,
"loss": 0.4482,
"step": 103
},
{
"epoch": 2.6666666666666665,
"grad_norm": 0.37817803025245667,
"learning_rate": 4.322727117869951e-08,
"loss": 0.4978,
"step": 104
},
{
"epoch": 2.6923076923076925,
"grad_norm": 0.39025190472602844,
"learning_rate": 3.734784976300165e-08,
"loss": 0.5465,
"step": 105
},
{
"epoch": 2.717948717948718,
"grad_norm": 0.3911699652671814,
"learning_rate": 3.188256468013139e-08,
"loss": 0.4831,
"step": 106
},
{
"epoch": 2.7435897435897436,
"grad_norm": 0.34893131256103516,
"learning_rate": 2.6836308100417872e-08,
"loss": 0.4515,
"step": 107
},
{
"epoch": 2.769230769230769,
"grad_norm": 0.3498555123806,
"learning_rate": 2.2213597106929605e-08,
"loss": 0.4528,
"step": 108
},
{
"epoch": 2.7948717948717947,
"grad_norm": 0.3860410749912262,
"learning_rate": 1.8018569652073378e-08,
"loss": 0.492,
"step": 109
},
{
"epoch": 2.8205128205128203,
"grad_norm": 0.26638758182525635,
"learning_rate": 1.4254980853566246e-08,
"loss": 0.3204,
"step": 110
},
{
"epoch": 2.8461538461538463,
"grad_norm": 0.4179665148258209,
"learning_rate": 1.0926199633097154e-08,
"loss": 0.5362,
"step": 111
},
{
"epoch": 2.871794871794872,
"grad_norm": 0.375400185585022,
"learning_rate": 8.035205700685165e-09,
"loss": 0.5113,
"step": 112
},
{
"epoch": 2.8974358974358974,
"grad_norm": 0.42177173495292664,
"learning_rate": 5.5845868874357385e-09,
"loss": 0.4808,
"step": 113
},
{
"epoch": 2.9230769230769234,
"grad_norm": 0.3527701795101166,
"learning_rate": 3.5765368290813223e-09,
"loss": 0.414,
"step": 114
},
{
"epoch": 2.948717948717949,
"grad_norm": 0.3130120635032654,
"learning_rate": 2.0128530023804656e-09,
"loss": 0.3637,
"step": 115
},
{
"epoch": 2.9743589743589745,
"grad_norm": 0.3594694435596466,
"learning_rate": 8.949351161324225e-10,
"loss": 0.4558,
"step": 116
},
{
"epoch": 3.0,
"grad_norm": 0.44394877552986145,
"learning_rate": 2.2378385824833866e-10,
"loss": 0.6395,
"step": 117
},
{
"epoch": 3.0,
"step": 117,
"total_flos": 3.3860293178247086e+18,
"train_loss": 0.5216725878735893,
"train_runtime": 16633.6833,
"train_samples_per_second": 0.449,
"train_steps_per_second": 0.007
}
],
"logging_steps": 1,
"max_steps": 117,
"num_input_tokens_seen": 0,
"num_train_epochs": 3,
"save_steps": 500,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 3.3860293178247086e+18,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}

3
training_args.bin Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:a8d439c8fb74afb34a05a9e122de5845863e92a5f1a5be959fe2912abf2c5f27
size 8145

BIN
training_loss.png Normal file

Binary file not shown.

After

Width:  |  Height:  |  Size: 60 KiB

1
vocab.json Normal file

File diff suppressed because one or more lines are too long