Skip to content

Commit ec90208

Browse files
committed
Add scripts to the README.md
1 parent b980b8a commit ec90208

File tree

2 files changed

+23
-2
lines changed

2 files changed

+23
-2
lines changed

examples/README.md

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ The SFT algorithm supports training language models on supervised datasets with
2727
- [SFT with Llama 3.1 8B](scripts/sft_llama_example.py) - Single-node multi-GPU training example with Llama 3.1 8B Instruct
2828
- [SFT with Phi 4 Mini](scripts/sft_phi_example.py) - Single-node multi-GPU training example with Phi 4 Mini Instruct
2929
- [SFT with GPT-OSS 20B](scripts/sft_gpt_oss_example.py) - Single-node multi-GPU training example with GPT-OSS 20B
30+
- [SFT with Granite 3.3 8B](scripts/sft_granite_example.py) - Single-node multi-GPU training example with Granite 3.3 8B Instruct
3031

3132
**Quick Example:**
3233
```python
@@ -60,6 +61,7 @@ The OSFT algorithm supports continual training of pre-trained or instruction-tun
6061
- [OSFT with Llama 3.1 8B](scripts/osft_llama_example.py) - Single-node multi-GPU training example with Llama 3.1 8B Instruct
6162
- [OSFT with Phi 4 Mini](scripts/osft_phi_example.py) - Single-node multi-GPU training example with Phi 4 Mini Instruct
6263
- [OSFT with GPT-OSS 20B](scripts/osft_gpt_oss_example.py) - Single-node multi-GPU training example with GPT-OSS 20B
64+
- [OSFT with Granite 3.3 8B](scripts/osft_granite_example.py) - Single-node multi-GPU training example with Granite 3.3 8B Instruct
6365
- [OSFT Continual Learning Example](scripts/osft_continual_learning_example.py) - Example script demonstrating continual learning without catastrophic forgetting
6466

6567
**Quick Example:**
@@ -101,6 +103,25 @@ estimate(training_method='osft',
101103
)
102104
```
103105

106+
### Model Interpolation (Experimental / In-Development)
107+
108+
training_hub has a utility for merging two checkpoints of the same model into one with linear interpolation.
109+
110+
**Script:**
111+
- [interpolator.py](scripts/interpolator.py) - Python script for model interpolation
112+
113+
**Command-Line Example:**
114+
```bash
115+
python interpolator.py --model-path ibm-granite/granite-3.3-8b-instruct --trained-model-path /path/to/checkpoint
116+
```
117+
118+
**Python Example:**
119+
```python
120+
from interpolator import interpolate_models
121+
122+
interpolate_models("ibm-granite/granite-3.3-8b-instruct", "/path/to/checkpoint")
123+
```
124+
104125
## Getting Started
105126

106127
1. **For detailed parameter documentation**: Check the relevant guide in `docs/`

examples/scripts/interpolator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
"""
2-
A simple model interpolation utility.
2+
Simple Model Interpolator
33
4-
This takes two checkpoints of the same model, and outputs a merged checkpoint with the linear interpolation.
4+
The script takes two checkpoints of the same model and outputs a merged checkpoint with linear interpolation.
55
66
Example usage:
77
python interpolator.py \\

0 commit comments

Comments
 (0)