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

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

About

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

Topics

Resources

Stars

Watchers

Forks

Packages

No packages published