#215 · Primary category: Deep Learning Frameworks

mergoo

artificial-intelligence fine-tuning generative-ai large-language-models llm lora merge mixture-of-adapters mixture-of-experts multi-model nlp open-source transformers

A library for easily merging multiple LLM experts, and efficiently train the merged LLM.

Project last updated:08/26/24

GitHub Stars

516

Forks

33

Contributors

7

License

LGPL-3.0

Why we included this project

Most teams with several fine-tuned checkpoints of the same base model would like the strengths of each without paying the cost of running them side by side. Mergoo handles that by routing Llama, Mistral, Phi3, or BERT checkpoints through a single network, and you can then fine-tune either the router alone or the whole merged model with the standard Hugging Face trainer APIs. It covers the two setups that actually come up in practice: merging fully fine-tuned experts into a mixture-of-experts stack, or putting a learned router on top of existing LoRA adapters. The library installs with pip, works with PEFT and SFTrainer, and ships notebooks for each merge method, which keeps the first prototype from turning into a routing-layer research project. If you are comparing model-composition strategies, configurable expert counts and router layers are built in.

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