← all repositories

AnswerDotAI/fsdp_qlora

A training framework enabling fine-tuning of large language models using quantized LoRA and fully-sharded data parallelism.

1.5k stars Jupyter Notebook ML FrameworksLanguage Models
fsdp_qlora
Velocity · 7d
+1.8
★ / day
Trend
steady
star history

This repository provides scripts and integration for training large language models with QLoRA (Quantized Low-Rank Adaptation) combined with FSDP (Fully Sharded Data Parallel). It enables efficient fine-tuning of models like Llama-2 70B on limited GPU memory by combining 4-bit quantization with gradient checkpointing and CPU offloading. The project integrates with Axolotl training framework and uses libraries like bitsandbytes and Hugging Face transformers.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.