Add fine-tuning module - #13

Merged

fingriffin merged 71 commits into

mainfrom

feature/finetuning

Jun 10, 2026

Commits on Apr 10, 2026

feat: add pilot sweep
fin-griffin
committed
626d91b
feat: add README
fin-griffin
committed
530f45f
chore: update README to select single gpu
fin-griffin
committed
1b75325

Commits on Apr 13, 2026

chore: migrate all sweep deps to uv
fin-griffin
committed
dfa44e8
chore: add hf cli
fin-griffin
committed
c411fdb
chore: update README.md
fin-griffin
committed
7df07a6
chore: add typer as dep
fin-griffin
committed
61e88f2
chore: update axolotl dep to download from source
fin-griffin
committed
2aef161
chore: update eval strategy to be a string rather than reserved boolean
fin-griffin
committed
1487684
fix: update chat template during inference
fin-griffin
committed
38ac129
fix: update infer.py to manually compute loss
fin-griffin
committed
87d23a8
fix: force inference to pin to same GPU as training
fin-griffin
committed
4caf6a9
chore: add vllm
fin-griffin
committed
4c9122d
chore: add bnb
fin-griffin
committed
8a2b5ba
fix: migrate to slow tokenizer for vllm
fin-griffin
committed
89bd4c0
fix: migrate to torch batched inference
fin-griffin
committed
f70bfc9
chore: fix triton kernels
fin-griffin
committed
e580e04
chore(deps): update vllm
fin-griffin
committed
b005b67
feat: migrate back to vLLM
fin-griffin
committed
7153cc1
chore: remove vllm
fin-griffin
committed
e42203b

Commits on Apr 14, 2026

chore: gitignore pilot results parquet
fin-griffin
committed
7d0a280
chore: add sweep results + logs to gitignore
fin-griffin
committed
24fe3ac
feat: migrate to vllm
fin-griffin
committed
59bb2ad
chore: fix run
fin-griffin
committed
6290698
fix 3
fin-griffin
committed
0f2ddd7
chore anotehr fix
fin-griffin
committed
a3357c0
jdkansjn
fin-griffin
committed
11eef81
ijas
fin-griffin
committed
f069241
chore: downgrade vllm
fin-griffin
committed
40ab156
fix transformers version
fin-griffin
committed
75fdb01
chore
fin-griffin
committed
b0b878f
migrate to online vllm
fin-griffin
committed
042dd16
update readme
fin-griffin
committed
5e24704
1
fin-griffin
committed
733a9ea
refactor(sweep): replace vLLM inference with in-process TrainerCallback
fin-griffin
committed
54213f6
fix(sweep): log sample completions at each epoch end
fin-griffin
committed
727a54c
docs(sweep): update README to remove vLLM setup, reflect callback approach
fin-griffin
committed
c1efff5
fix(sweep): add repo root to PYTHONPATH so axolotl can import EvalCompletionsCallback
fin-griffin
committed
11fde90
build: upgrade axolotl pin from v0.16.1 to main for callbacks YAML support
fin-griffin
committed
2218301
fix(sweep): switch to axolotl plugins API for callback registration
fin-griffin
committed
1fd9cbf

Commits on Apr 15, 2026

fix: axolotl from source
fin-griffin
committed
e3b3d24
fix: tokenizer = None err
fin-griffin
committed
5c28180
fix:
fin-griffin
committed
2ac9558
chore: remove padding
fin-griffin
committed
5cedd8a
chore: temperature -> 0.7
fin-griffin
committed
027794a
add batching
fin-griffin
committed
eda3fe2
chore: update default devices
fin-griffin
committed
4f4dd71
chore: update wandb project
fin-griffin
committed
8468602
fix eval loss logging
fin-griffin
committed
4c091e3

Commits on May 21, 2026

feat(docs): add docs for data, evals and stylometry
fin-griffin
committed
2040113
Merge branch 'main' into feature/pilot-sweep
fin-griffin
committed
9ea05b1

Commits on May 22, 2026

chore: update gitignore
fin-griffin
committed
6fe6287
chore: add configs and run dirs
fin-griffin
committed
81d76d1
feat(data): add local dataset loading to VoiceDataset
fin-griffin
committed
f5893da
feat(fine-tuning): add fine-tuning module for single runs and grid search
fin-griffin
committed
a8b7d59
feat(fine-tuning): add run directory management
fin-griffin
committed
17a8fbf
feat(axolotl): add custom callback
fin-griffin
committed
302e8ae
feat(fine-tuning): add orchestration module
fin-griffin
committed
609851d
feat(fine-tuning): add cli
fin-griffin
committed
0bc728e
chore: remove wandb post-hoc artefact upload
fin-griffin
committed
eca7f27
chore: remove unicode escaping in axolotl custom callback
fin-griffin
committed
2d384b1
fix: fix PYTHONPATH handling in axolotl subprocess
fin-griffin
committed
dd2ac29

Commits on Jun 9, 2026

feat(stylometry): remove number of tokens/words as a stylometric metric
fin-griffin
committed
99cf6c0

Commits on Jun 10, 2026