|
|
--- |
|
|
base_model: |
|
|
- allenai/Olmo-3-1025-7B |
|
|
- allenai/Olmo-3-7B-RL-Zero-Math |
|
|
- allenai/Olmo-3-7B-RL-Zero-Code |
|
|
- allenai/Olmo-3-7B-Think-SFT |
|
|
library_name: transformers |
|
|
tags: |
|
|
- mergekit |
|
|
- merge |
|
|
|
|
|
--- |
|
|
# merged-model |
|
|
|
|
|
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 [allenai/Olmo-3-1025-7B](https://huggingface.co/allenai/Olmo-3-1025-7B) as a base. |
|
|
|
|
|
### Models Merged |
|
|
|
|
|
The following models were included in the merge: |
|
|
* [allenai/Olmo-3-7B-RL-Zero-Math](https://huggingface.co/allenai/Olmo-3-7B-RL-Zero-Math) |
|
|
* [allenai/Olmo-3-7B-RL-Zero-Code](https://huggingface.co/allenai/Olmo-3-7B-RL-Zero-Code) |
|
|
* [allenai/Olmo-3-7B-Think-SFT](https://huggingface.co/allenai/Olmo-3-7B-Think-SFT) |
|
|
|
|
|
### Configuration |
|
|
|
|
|
The following YAML configuration was used to produce this model: |
|
|
|
|
|
```yaml |
|
|
# Task arithmetic merge: Apply Math+Code task vectors to Think-SFT |
|
|
# |
|
|
# Mathematical formulation: |
|
|
# output = Think-SFT + 0.5*(RL-Zero-Math - base) + 0.5*(RL-Zero-Code - base) |
|
|
# |
|
|
# This is achieved by treating Think-SFT as a model with weight=1.0: |
|
|
# output = base + 1.0*(Think-SFT - base) + 0.5*(Math - base) + 0.5*(Code - base) |
|
|
# |
|
|
# Usage: |
|
|
# modal run modal_merge.py --config examples/olmo-think-math-code.yaml --hf-repo pmahdavi/Olmo-3-7B-Think-Math-Code |
|
|
|
|
|
merge_method: task_arithmetic |
|
|
base_model: allenai/Olmo-3-1025-7B |
|
|
models: |
|
|
- model: allenai/Olmo-3-7B-Think-SFT |
|
|
parameters: |
|
|
weight: 1.0 |
|
|
- model: allenai/Olmo-3-7B-RL-Zero-Math |
|
|
parameters: |
|
|
weight: 0.5 |
|
|
- model: allenai/Olmo-3-7B-RL-Zero-Code |
|
|
parameters: |
|
|
weight: 0.5 |
|
|
dtype: bfloat16 |
|
|
|
|
|
``` |
|
|
|