Merge Diffusion Tool

Объединение диффузионного инструмента

Инструмент Diffusion Merge -это решение с открытым исходным кодом, разработанное Enhanceai.art , обеспечивающее бесшовные способы смешивания моделей Lora, интегрировать LORA в контрольные точки и объединять стабильные контрольные точки диффузии. Улучшите свои рабочие процессы ИИ с помощью этого мощного инструмента объединения, предназначенного для поддержки широкого диапазона диффузионных моделей, таких как Flux Dev , Flux Schnell , стабильная диффузия 1.5 , SD2 , SD3 и SDXL .

Откройте для себя более продвинутые инструменты искусственного интеллекта в Enhanceai.art.

Функции

• Объедините две модели LORA с регулируемыми соотношениями смешивания.
• Интегрируйте LORA в контрольную точку для повышения производительности модели.
• Объедините две контрольные точки с помощью пользовательских соотношений смешивания.
• Полная поддержка формата .safetensors , обеспечивающая эффективную и безопасную обработку.

Ключевые преимущества:

• Опропит свой процесс слияния модели ИИ.
• Построен с гибкостью, чтобы удовлетворить различные форматы моделей ИИ.
• Полностью открытый источник с поддержкой сообщества через Discord.

---

Установка

Чтобы настроить инструмент, следуйте этим шагам:

1. Клонировать репозиторий:

   git clone https://github.com/itspranavajay/Merge-Diffusion-Tool.git
   cd merge-diffusion-tool

2. Установите требуемые зависимости:

   pip install -r requirements.txt

Зависимости:

• Python 3.8+
• Pytorch для глубокого обучения.
• Библиотека Safetensors для обработки файлов модели.

---

Использование

Инструмент диффузии слияния предоставляет три основных операция:

1. Объединить две модели LORA
2. Объединить LORA в контрольно -пропускной пункт
3. Объединить две контрольно -пропускные пункты

Аргументы командной строки:

Аргумент	Описание
--operation	Выберите операцию: merge_loras , merge_lora_checkpoint , merge_checkpoints
--model1	Путь к первой модели файл.
--model2	Путь ко второму файлу модели (необязательно для Lora на контрольную точку).
--output	Выходной путь для объединенной модели.
--alpha	Соотношение смеси для слияния моделей (по умолчанию: 0,5).
--merge_weight	Вес для объединения LORA с контрольной точкой (по умолчанию: 0,5).

Пример 1: объединение двух моделей LORA

python main.py --operation merge_loras --model1 lora1.safetensors --model2 lora2.safetensors --output output_lora.safetensors --alpha 0.7

Эта команда объединяет lora1.safetensors и lora2.safetensors , с 70% вкладом lora1 .

Пример 2: Слияние Лоры в контрольно -пропускной пункт

python main.py --operation merge_lora_checkpoint --model1 lora_model.safetensors --model2 checkpoint_model.safetensors --output output_checkpoint.safetensors --merge_weight 0.6

В этом примере модель LORA объединяется в контрольную точку с 60% влиянием.

Пример 3: Слияние двух контрольных точек

python main.py --operation merge_checkpoints --model1 checkpoint1.safetensors --model2 checkpoint2.safetensors --output output_checkpoint.safetensors --alpha 0.5

Объединяет обе модели контрольных точек в смеси 50-50.

---

Поддерживаемые форматы

• .safetensors : обеспечивает безопасное, оптимизированное хранение и обработку моделей искусственного интеллекта.

---

Пожертвование Enhanceai.art

? Если вы обнаружите, что этот инструмент полезным, пожалуйста, рассмотрите возможность поддержки разработки, проверив наши планы ценообразования на enfanceai.art/pracing. Покупая любой план, вы не только поддерживаете будущие проекты с открытым исходным кодом, но вы также разблокируете многие эксклюзивные функции искусственного интеллекта , которые значительно улучшат себя рабочими процессами ИИ!

---

Поддерживать

Для получения помощи присоединяйтесь к нашему серверу Discord, чтобы общаться с сообществом и разработчиками.

---

Внося

Мы приветствуем вклад! Не стесняйтесь открывать проблемы, отправлять запросы на привлечение или предложить функции, которые улучшат этот инструмент. Все вклады высоко ценится.