thlurte's picture
Upload folder using huggingface_hub
fa7906f verified
|
Raw
History Blame Contribute Delete
1.26 kB
---
base_model:
- Qwen/Qwen2.5-1.5B
- Qwen/Qwen2.5-Math-1.5B
- Qwen/Qwen2.5-Coder-1.5B
library_name: transformers
tags:
- mergekit
- merge
---
# Qwen2.5-CodeMath-1.5B-TaskArith
This is a merge of pre-trained language models created using [mergekit](https://github.com/cg123/mergekit).
## Merge Details
### Merge Method
This model was merged using the [Task Arithmetic](https://arxiv.org/abs/2212.04089) merge method using [Qwen/Qwen2.5-1.5B](https://huggingface.co/Qwen/Qwen2.5-1.5B) as a base.
### Models Merged
The following models were included in the merge:
* [Qwen/Qwen2.5-Math-1.5B](https://huggingface.co/Qwen/Qwen2.5-Math-1.5B)
* [Qwen/Qwen2.5-Coder-1.5B](https://huggingface.co/Qwen/Qwen2.5-Coder-1.5B)
### Configuration
The following YAML configuration was used to produce this model:
```yaml
base_model: Qwen/Qwen2.5-1.5B
dtype: bfloat16
merge_method: task_arithmetic
modules:
default:
slices:
- sources:
- layer_range: [0, 28]
model: Qwen/Qwen2.5-Coder-1.5B
parameters:
weight: 1.0
- layer_range: [0, 28]
model: Qwen/Qwen2.5-Math-1.5B
parameters:
weight: 0.25
- layer_range: [0, 28]
model: Qwen/Qwen2.5-1.5B
tokenizer_source: Qwen/Qwen2.5-Coder-1.5B
```