abhid1234/qwen-0.5b-tool-agent-grpo
1
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

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

Browse Source 
Model: abhid1234/qwen-0.5b-tool-agent-grpo
Source: Original Platform
This commit is contained in:
ModelHub XC
2026-04-28 07:59:12 +08:00
commit 2e4a8d6a83
11 changed files with 1863 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
.gitattributes vendored Normal file
View File

@@ -0,0 +1,36 @@
*.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

199
README.md Normal file
View File

@@ -0,0 +1,199 @@
---
library_name: transformers
tags: []
---
# Model Card for Model ID
<!-- Provide a quick summary of what the model is/does. -->
## Model Details
### Model Description
<!-- Provide a longer summary of what this model is. -->
This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.
- **Developed by:** [More Information Needed]
- **Funded by [optional]:** [More Information Needed]
- **Shared by [optional]:** [More Information Needed]
- **Model type:** [More Information Needed]
- **Language(s) (NLP):** [More Information Needed]
- **License:** [More Information Needed]
- **Finetuned from model [optional]:** [More Information Needed]
### Model Sources [optional]
<!-- Provide the basic links for the model. -->
- **Repository:** [More Information Needed]
- **Paper [optional]:** [More Information Needed]
- **Demo [optional]:** [More Information Needed]
## Uses
<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
### Direct Use
<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
[More Information Needed]
### Downstream Use [optional]
<!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
[More Information Needed]
### Out-of-Scope Use
<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
[More Information Needed]
## Bias, Risks, and Limitations
<!-- This section is meant to convey both technical and sociotechnical limitations. -->
[More Information Needed]
### Recommendations
<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
## How to Get Started with the Model
Use the code below to get started with the model.
[More Information Needed]
## Training Details
### Training Data
<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
[More Information Needed]
### Training Procedure
<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
#### Preprocessing [optional]
[More Information Needed]
#### Training Hyperparameters
- **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
#### Speeds, Sizes, Times [optional]
<!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
[More Information Needed]
## Evaluation
<!-- This section describes the evaluation protocols and provides the results. -->
### Testing Data, Factors & Metrics
#### Testing Data
<!-- This should link to a Dataset Card if possible. -->
[More Information Needed]
#### Factors
<!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
[More Information Needed]
#### Metrics
<!-- These are the evaluation metrics being used, ideally with a description of why. -->
[More Information Needed]
### Results
[More Information Needed]
#### Summary
## Model Examination [optional]
<!-- Relevant interpretability work for the model goes here -->
[More Information Needed]
## Environmental Impact
<!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
- **Hardware Type:** [More Information Needed]
- **Hours used:** [More Information Needed]
- **Cloud Provider:** [More Information Needed]
- **Compute Region:** [More Information Needed]
- **Carbon Emitted:** [More Information Needed]
## Technical Specifications [optional]
### Model Architecture and Objective
[More Information Needed]
### Compute Infrastructure
[More Information Needed]
#### Hardware
[More Information Needed]
#### Software
[More Information Needed]
## Citation [optional]
<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
**BibTeX:**
[More Information Needed]
**APA:**
[More Information Needed]
## Glossary [optional]
<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
[More Information Needed]
## More Information [optional]
[More Information Needed]
## Model Card Authors [optional]
[More Information Needed]
## Model Card Contact
[More Information Needed]

412
artifacts/eval_results.json Normal file
View File

@@ -0,0 +1,412 @@
{
"step": 15,
"scenarios_path": "data/scenarios_val.jsonl",
"num_generations": 8,
"total_scenarios": 50,
"total_rollouts": 400,
"successes": 18,
"accuracy_pct": 4.5,
"avg_reward": -1.8850000000000002,
"per_scenario": [
{
"scenario_index": 0,
"task": "Convert 98 kg to lbs.",
"mean_reward": 3.125,
"max_reward": 4.0,
"success_count": 7,
"total_attempts": 8
},
{
"scenario_index": 1,
"task": "What is the speed of light?",
"mean_reward": -2.5,
"max_reward": -2.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 2,
"task": "What is the distance from Earth to the Sun in km in miles?",
"mean_reward": 0.5625,
"max_reward": 2.5,
"success_count": 1,
"total_attempts": 8
},
{
"scenario_index": 3,
"task": "What is 441 plus 23?",
"mean_reward": 2.25,
"max_reward": 4.0,
"success_count": 1,
"total_attempts": 8
},
{
"scenario_index": 4,
"task": "Convert 62 kg to lbs.",
"mean_reward": -3.0,
"max_reward": -3.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 5,
"task": "Which is hotter right now, London or Mumbai?",
"mean_reward": -2.875,
"max_reward": -2.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 6,
"task": "What is 185 plus 89?",
"mean_reward": -3.0,
"max_reward": -3.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 7,
"task": "What's the weather like in Dubai?",
"mean_reward": -1.375,
"max_reward": 4.0,
"success_count": 2,
"total_attempts": 8
},
{
"scenario_index": 8,
"task": "What is the population of Germany divided by its area in km2?",
"mean_reward": -2.041666666666667,
"max_reward": 1.666666666666666,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 9,
"task": "What is the boiling point of water?",
"mean_reward": -3.0,
"max_reward": -3.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 10,
"task": "Which is hotter right now, London or Mumbai?",
"mean_reward": -2.0,
"max_reward": -2.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 11,
"task": "What is the population of India divided by its area in km2?",
"mean_reward": -2.125,
"max_reward": -2.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 12,
"task": "What is India's population density in people per square mile?",
"mean_reward": -1.25,
"max_reward": 1.333333333333333,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 13,
"task": "What is the tallest mountain?",
"mean_reward": -2.875,
"max_reward": -2.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 14,
"task": "What is the distance from Earth to the Sun in km in miles?",
"mean_reward": -1.875,
"max_reward": 1.5,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 15,
"task": "What is the population of Japan divided by its area in km2?",
"mean_reward": -2.875,
"max_reward": -2.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 16,
"task": "What is Germany's population density in people per square mile?",
"mean_reward": -3.0,
"max_reward": -3.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 17,
"task": "Convert 74 kg to lbs.",
"mean_reward": -3.0,
"max_reward": -3.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 18,
"task": "Which is hotter right now, Paris or Cairo?",
"mean_reward": -3.0,
"max_reward": -3.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 19,
"task": "What is India's population density in people per square mile?",
"mean_reward": -1.5,
"max_reward": 1.333333333333333,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 20,
"task": "Which country has a larger population, France or Brazil?",
"mean_reward": -3.0,
"max_reward": -3.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 21,
"task": "Convert 64 kg to lbs.",
"mean_reward": -3.0,
"max_reward": -3.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 22,
"task": "Which country has a larger population, Japan or India?",
"mean_reward": -1.5625,
"max_reward": 3.0,
"success_count": 2,
"total_attempts": 8
},
{
"scenario_index": 23,
"task": "What is the GDP of Japan?",
"mean_reward": -2.75,
"max_reward": -2.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 24,
"task": "What is the population of France divided by its area in km2?",
"mean_reward": -2.875,
"max_reward": -2.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 25,
"task": "What is France's population density in people per square mile?",
"mean_reward": -2.5,
"max_reward": 1.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 26,
"task": "Convert 26 kg to lbs.",
"mean_reward": -3.0,
"max_reward": -3.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 27,
"task": "What is 660 times 87?",
"mean_reward": 1.0,
"max_reward": 4.0,
"success_count": 1,
"total_attempts": 8
},
{
"scenario_index": 28,
"task": "What is the boiling point of water?",
"mean_reward": -2.5,
"max_reward": -2.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 29,
"task": "What is the population of Germany divided by its area in km2?",
"mean_reward": -1.125,
"max_reward": 2.333333333333333,
"success_count": 1,
"total_attempts": 8
},
{
"scenario_index": 30,
"task": "Convert 40 kg to lbs.",
"mean_reward": -3.0,
"max_reward": -3.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 31,
"task": "What is the speed of light?",
"mean_reward": -2.375,
"max_reward": -2.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 32,
"task": "How old was Guido van Rossum in 2024?",
"mean_reward": -2.875,
"max_reward": -2.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 33,
"task": "Which is hotter right now, Paris or Dubai?",
"mean_reward": -3.0,
"max_reward": -3.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 34,
"task": "Which is hotter right now, Tokyo or Dubai?",
"mean_reward": -2.875,
"max_reward": -2.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 35,
"task": "Which is hotter right now, London or Cairo?",
"mean_reward": -2.375,
"max_reward": -2.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 36,
"task": "What is the value of pi?",
"mean_reward": -2.125,
"max_reward": -2.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 37,
"task": "What is the population of Japan divided by its area in km2?",
"mean_reward": -1.6666666666666667,
"max_reward": 1.666666666666666,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 38,
"task": "What is the temperature in London in Fahrenheit?",
"mean_reward": -1.375,
"max_reward": 1.5,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 39,
"task": "What is 464 plus 30?",
"mean_reward": -1.75,
"max_reward": 2.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 40,
"task": "Which country has a larger population, France or India?",
"mean_reward": -2.1875,
"max_reward": 2.5,
"success_count": 1,
"total_attempts": 8
},
{
"scenario_index": 41,
"task": "What is the distance from Earth to the Sun in km in miles?",
"mean_reward": -1.625,
"max_reward": 2.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 42,
"task": "What is the tallest mountain?",
"mean_reward": -1.125,
"max_reward": 2.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 43,
"task": "What is the temperature in London in Fahrenheit?",
"mean_reward": 0.6875,
"max_reward": 2.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 44,
"task": "What is 496 minus 24?",
"mean_reward": 1.0,
"max_reward": 2.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 45,
"task": "What's the weather like in Cairo?",
"mean_reward": -1.25,
"max_reward": 4.0,
"success_count": 2,
"total_attempts": 8
},
{
"scenario_index": 46,
"task": "What is the tallest mountain?",
"mean_reward": -2.5,
"max_reward": -2.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 47,
"task": "What is India's population density in people per square mile?",
"mean_reward": -1.7916666666666667,
"max_reward": 1.333333333333333,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 48,
"task": "What is the GDP of France?",
"mean_reward": -2.625,
"max_reward": -2.0,
"success_count": 0,
"total_attempts": 8
},
{
"scenario_index": 49,
"task": "How old was Guido van Rossum in 2024?",
"mean_reward": -2.75,
"max_reward": -2.0,
"success_count": 0,
"total_attempts": 8
}
]
}

16
artifacts/reward_curve.txt Normal file
View File

@@ -0,0 +1,16 @@
Avg reward: -0.208 | Avg tools/rollout: 0.9 | groups with variance: 4/4
Avg reward: 1.969 | Avg tools/rollout: 1.0 | groups with variance: 1/4
Avg reward: 0.854 | Avg tools/rollout: 1.0 | groups with variance: 4/4
Avg reward: 1.193 | Avg tools/rollout: 0.9 | groups with variance: 3/4
Avg reward: -2.094 | Avg tools/rollout: 0.8 | groups with variance: 3/4
Avg reward: 0.505 | Avg tools/rollout: 0.9 | groups with variance: 4/4
Avg reward: -0.141 | Avg tools/rollout: 0.8 | groups with variance: 4/4
Avg reward: -0.797 | Avg tools/rollout: 0.9 | groups with variance: 4/4
Avg reward: 0.307 | Avg tools/rollout: 0.9 | groups with variance: 3/4
Avg reward: -1.125 | Avg tools/rollout: 1.0 | groups with variance: 1/4
Avg reward: -1.359 | Avg tools/rollout: 0.9 | groups with variance: 4/4
Avg reward: 0.484 | Avg tools/rollout: 1.0 | groups with variance: 3/4
Avg reward: -0.073 | Avg tools/rollout: 0.9 | groups with variance: 4/4
Avg reward: 1.740 | Avg tools/rollout: 1.0 | groups with variance: 3/4
Avg reward: 0.635 | Avg tools/rollout: 1.0 | groups with variance: 3/4
Avg reward: 1.615 | Avg tools/rollout: 0.9 | groups with variance: 2/4

1054
artifacts/training.log Normal file
View File

File diff suppressed because one or more lines are too long

54
chat_template.jinja Normal file
View File

@@ -0,0 +1,54 @@
{%- if tools %}
{{- '<|im_start|>system\n' }}
{%- if messages[0]['role'] == 'system' %}
{{- messages[0]['content'] }}
{%- else %}
{{- 'You are Qwen, created by Alibaba Cloud. You are a helpful assistant.' }}
{%- endif %}
{{- "\n\n# 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' }}
{%- else %}
{{- '<|im_start|>system\nYou are Qwen, created by Alibaba Cloud. You are a helpful assistant.<|im_end|>\n' }}
{%- endif %}
{%- endif %}
{%- for message in messages %}
{%- if (message.role == "user") or (message.role == "system" and not loop.first) or (message.role == "assistant" and not message.tool_calls) %}
{{- '<|im_start|>' + message.role + '\n' + message.content + '<|im_end|>' + '\n' }}
{%- elif message.role == "assistant" %}
{{- '<|im_start|>' + message.role }}
{%- if message.content %}
{{- '\n' + message.content }}
{%- endif %}
{%- for tool_call in message.tool_calls %}
{%- if tool_call.function is defined %}
{%- set tool_call = tool_call.function %}
{%- endif %}
{{- '\n<tool_call>\n{"name": "' }}
{{- tool_call.name }}
{{- '", "arguments": ' }}
{{- tool_call.arguments | tojson }}
{{- '}\n</tool_call>' }}
{%- endfor %}
{{- '<|im_end|>\n' }}
{%- elif message.role == "tool" %}
{%- if (loop.index0 == 0) or (messages[loop.index0 - 1].role != "tool") %}
{{- '<|im_start|>user' }}
{%- endif %}
{{- '\n<tool_response>\n' }}
{{- message.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' }}
{%- endif %}

57
config.json Normal file
View File

@@ -0,0 +1,57 @@
{
"architectures": [
"Qwen2ForCausalLM"
],
"attention_dropout": 0.0,
"bos_token_id": 151643,
"dtype": "bfloat16",
"eos_token_id": 151645,
"hidden_act": "silu",
"hidden_size": 896,
"initializer_range": 0.02,
"intermediate_size": 4864,
"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"
],
"max_position_embeddings": 32768,
"max_window_layers": 21,
"model_type": "qwen2",
"num_attention_heads": 14,
"num_hidden_layers": 24,
"num_key_value_heads": 2,
"pad_token_id": null,
"rms_norm_eps": 1e-06,
"rope_parameters": {
"rope_theta": 1000000.0,
"rope_type": "default"
},
"sliding_window": null,
"tie_word_embeddings": true,
"transformers_version": "5.2.0",
"use_cache": true,
"use_sliding_window": false,
"vocab_size": 151936
}

14
generation_config.json Normal file
View File

@@ -0,0 +1,14 @@
{
"bos_token_id": 151643,
"do_sample": true,
"eos_token_id": [
151645,
151643
],
"pad_token_id": 151643,
"repetition_penalty": 1.1,
"temperature": 0.7,
"top_k": 20,
"top_p": 0.8,
"transformers_version": "5.2.0"
}

3
model.safetensors Normal file
View File

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

3
tokenizer.json Normal file
View File

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

15
tokenizer_config.json Normal file
View File

@@ -0,0 +1,15 @@
{
"add_prefix_space": false,
"backend": "tokenizers",
"bos_token": null,
"clean_up_tokenization_spaces": false,
"eos_token": "<|im_end|>",
"errors": "replace",
"is_local": true,
"model_max_length": 32768,
"pad_token": "<|PAD_TOKEN|>",
"padding_side": "right",
"split_special_tokens": false,
"tokenizer_class": "Qwen2Tokenizer",
"unk_token": null
}