Python
5 stars
End-to-end distributed LLM fine-tuning pipeline: data curation → QLoRA + DPO training → TIES/DARE/SLERP model merging → evaluation → vLLM deployment. Built with DeepSpeed ZeRO-3, Flash Attention, Prometheus monitoring, and AWS spot instance infrastructure.
### Overview: This project is an end-to-end distributed LLM fine-tuning pipeline that includes data curation, QLoRA + DPO training, model merging, evaluation, and deployment. Built with DeepSpeed ZeRO-3, Flash Attention, Prometheus monitoring, and AWS spot instance infrastructure, it is adaptable for any HuggingFace-compatible model.
### Key features:
- **Data Curation**: Scrapes data from arXiv, GitHub, StackOverflow, HuggingFace, and custom URLs, cleans it using MinHash LSH and PII removal, and filters based on quality metrics.
- **Training**: Uses QLoRA + DPO training, DeepSpeed ZeRO-3, Flash Attention, and HuggingFace-compatible models.
- **Model Merging**: Combines models using TIES, DARE, and SLERP methods, with weights optimized for performance.
- **Evaluation**: Evaluates models using benchmarks like MMLU, LLM-as-judge, and statistical significance.
- **Deployment**: Deployed using Docker and AWS spot instances, with FastAPI and Terraform configurations.
### Tech stack: Python, DeepSpeed ZeRO-3, Flash Attention, Prometheus, Docker, AWS spot instances, FastAPI, Terraform.
### Why it matters: This pipeline is designed to streamline the fine-tuning, merging, and deployment of large language models, making it highly adaptable and efficient for various HuggingFace-compatible models.