Skip to content

LLM Finetuning with Axolotl with decent defaults + Optional TrueFoundry Experiment Tracking Extension

Notifications You must be signed in to change notification settings

truefoundry/llm-finetune

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Important

Please prefer using commits from release tags. main branch is work in progress and may have partially working commits.

LLM Finetuning with Truefoundry

Test QLoRA w/ Deepspeed Stage 2

./sample_run.sh

TODO:

  • Setup C/I Tests
  • Track and publish VRAM and Speed benchmarks for popular models and GPUs

Generally we always try to optimize for memory footprint because that allows higher batch size and more gpu utilization Speedup is second priority but we take what we can easily get