Библиотека обучения на PyTorch с нативной поддержкой DTensor для дообучения и предобучения LLM, VLM и diffusion-моделей непосредственно из чекпойнтов Hugging Face.
arcee-ai/mergekit
Набор инструментов для слияния предобученных больших языковых моделей путём комбинирования их весов разными алгоритмами на CPU или GPU.
- Звёзды
- 7,439
- Форки
- 803
- Открытые issues
- 276
- Последний коммит
- 7 дней назад
Что это?
mergekit объединяет веса предобученных языковых моделей в одну без дополнительного обучения. Инструмент рассчитан на работу с данными, не помещающимися в память целиком: тензоры загружаются по требованию, поэтому сложные слияния можно выполнять на CPU или на GPU с объёмом видеопамяти всего 8 ГБ.
Параметры слияния задаются в YAML-файлах конфигурации. В конфиге выбирают метод слияния, исходные модели или срезы слоёв, а также параметры отдельных тензоров, например веса и плотность. Набор инструментов поддерживает широкий спектр методов, включая linear, SLERP, TIES, DARE, DELLA и Model Stock, а также слияние путём стекирования слоёв (passthrough), извлечение LoRA, сборку mixture-of-experts и согласование токенизаторов моделей с разными словарями.
Помимо основной команды `mergekit-yaml`, проект включает инструменты для многоэтапных сценариев слияния, объединения исходных чекпойнтов PyTorch и переноса токенизаторов. Также доступен Python API, который позволяет применять методы слияния к тензорам или state dict в памяти.
Ключевые возможности
- 01Множество методов слияния, включая linear, SLERP, TIES, DARE, DELLA и Model Stock
- 02Ленивая загрузка тензоров с запуском на CPU или GPU примерно с 8 ГБ видеопамяти
- 03Конфигурация через YAML с настройкой параметров отдельных тензоров
- 04Слияние путём стекирования слоёв (passthrough) и сборка mixture-of-experts
- 05Извлечение LoRA в виде адаптеров, совместимых с PEFT
- 06Согласование токенизаторов моделей с разными словарями
Сценарии использования
- Объединение нескольких дообученных чекпойнтов одной базовой модели в одну универсальную модель
- Создание экспериментальных frankenmerge-моделей путём стекирования слоёв разных моделей
- Подготовка черновых моделей для speculative decoding с помощью переноса токенизатора
- Извлечение LoRA-адаптеров из дообученных моделей относительно их базовых моделей
Быстрый старт
- 1
Клонируйте репозиторий
git clone https://github.com/arcee-ai/mergekit.git - 2
Перейдите в каталог проекта
cd mergekit - 3
Установите пакет в редактируемом режиме
pip install -e . - 4
Выполните слияние с файлом конфигурации YAML
mergekit-yaml path/to/your/config.yml ./output-model-directory
Что важно знать
- Распространяется по лицензии LGPL-3.0 — лицензии с авторским левом, условия которой распространяются и на изменения кода библиотеки.
- В README описана только установка из исходного кода с помощью pip install -e .; способ установки опубликованного пакета не указан.
Темы
Частые вопросы
- Что такое mergekit?
- Набор инструментов для слияния предобученных больших языковых моделей путём комбинирования их весов разными алгоритмами на CPU или GPU. mergekit объединяет веса предобученных языковых моделей в одну без дополнительного обучения. Инструмент рассчитан на работу с данными, не помещающимися в память целиком: тензоры загружаются по требованию, поэтому сложные слияния можно выполнять на CPU или на GPU с объёмом видеопамяти всего 8 ГБ.
- mergekit можно использовать бесплатно? Какая у него лицензия?
- mergekit распространяется с открытым исходным кодом под лицензией LGPL-3.0. Перед использованием в коммерческих продуктах проверьте условия лицензии.
- mergekit всё ещё поддерживается?
- Активно поддерживается. Последний коммит в mergekit — 31 окт. 2026 г..
- Для чего можно использовать mergekit?
- Объединение нескольких дообученных чекпойнтов одной базовой модели в одну универсальную модель Создание экспериментальных frankenmerge-моделей путём стекирования слоёв разных моделей Подготовка черновых моделей для speculative decoding с помощью переноса токенизатора Извлечение LoRA-адаптеров из дообученных моделей относительно их базовых моделей
- Как установить mergekit?
- Клонируйте репозиторий: git clone https://github.com/arcee-ai/mergekit.git Перейдите в каталог проекта: cd mergekit Установите пакет в редактируемом режиме: pip install -e . Выполните слияние с файлом конфигурации YAML: mergekit-yaml path/to/your/config.yml ./output-model-directory
- Где скачать mergekit?
- Исходный код и инструкции по установке — на GitHub: https://github.com/arcee-ai/mergekit
Описание создано ИИ на основе README проекта. Статистика GitHub обновляется ежедневно.
Похожие проекты
CLI на основе конфигурации для запуска и сравнения экспериментов по дообучению LLM: настройке промптов, моделей с открытым исходным кодом и LoRA, с тестами результатов.
Набор Agent Skills с инструкциями и скриптами для AI-агентов, которые выполняют задачи на Hugging Face Hub: обучение, оценку, работу с датасетами и Spaces.
SliceGPT сжимает трансформерные модели, включая LLM, после обучения, удаляя строки и столбцы матриц весов.
Полный пайплайн предобучения с нуля и SFT небольшой китайской модели в стиле LLaMA2, который можно запустить на одной видеокарте с 24 ГБ памяти.
Открытые большие языковые модели, дообученные для финансовых задач: анализа тональности, прогнозирования и тестов по финансовой обработке естественного языка.