#183 · Primary category: Deep Learning Frameworks

MFTCoder

customizable multi-model-support multi-task-fine-tuning multi-task-learning user-friendly

High Accuracy and efficiency multi-task fine-tuning framework for Code LLMs. This work has been accepted by KDD 2024.

Project last updated:12/30/24

GitHub Stars

711

Forks

69

Contributors

12

License

Apache-2.0

Why we included this project

MFTCoder takes a different route to fine-tuning code models: instead of adapting a model to one task at a time, it trains on several related code tasks together, which the project argues is both more accurate and more efficient. The framework offers two training paths, one built on PEFT and one on accelerate, and the accelerate path supports QLoRA with DeepSpeed ZeRO-3 or FSDP, so large models can be fine-tuned without a big GPU cluster. The latest release also adds preference alignment methods (DPO, RPO, ORPO) for steering output behavior rather than just teaching new tasks. Two peer-reviewed papers, one at KDD 2024 and one at EMNLP 2024, explain the multitask and convergence-balancing methods, which helps if you want the theory behind the code, not just a script to run.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category