Merge Diffusion Tool

Ferge Ferramenta de difusão

A Ferramenta de Difusão de Merge é uma solução de código aberto desenvolvido pelo ENFIANCEAI.ART , fornecendo maneiras perfeitas de misturar modelos Lora, integrar Lora nos pontos de verificação e mesclar pontos de verificação de difusão estável. Aprimore seus fluxos de trabalho de IA com esta poderosa ferramenta de fusão, projetada para suportar uma ampla gama de modelos de difusão como fluxo dev , fluxo schnell , difusão estável 1.5 , sd2 , sd3 e sdxl .

Descubra ferramentas de IA mais avançadas em Evenenceai.art.

Características

• Mesclar dois modelos Lora com taxas de mistura ajustáveis.
• Integre o LORA em um ponto de verificação para obter um desempenho aprimorado do modelo.
• Mesclar dois pontos de verificação usando taxas de mistura personalizadas.
• Suporte total para o formato .safetensors , garantindo manuseio eficiente e seguro.

Principais vantagens:

• Simplifique seu processo de fusão do modelo de IA.
• Construído com flexibilidade para atender a diversos formatos de modelo de IA.
• Completamente de código aberto com apoio da comunidade via Discord.

---

Instalação

Para configurar a ferramenta, siga estas etapas:

1. Clone o repositório:

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

2. Instale as dependências necessárias:

   pip install -r requirements.txt

Dependências:

• Python 3.8+
• Pytorch para operações de aprendizado profundo.
• Biblioteca SafeTetenss para manuseio de arquivos de modelo.

---

Uso

A ferramenta de difusão de mesclagem fornece três operações principais:

1. Mesclar dois modelos Lora
2. Merge Lora em um posto de controle
3. Mesclar dois pontos de verificação

Argumentos da linha de comando:

Argumento	Descrição
--operation	Selecione Operação: merge_loras , merge_lora_checkpoint , merge_checkpoints
--model1	Caminho para o primeiro arquivo de modelo.
--model2	Caminho para o segundo arquivo do modelo (opcional para Lora no ponto de verificação).
--output	Caminho de saída para o modelo mesclado.
--alpha	Razão de mistura para modelos de fusão (padrão: 0,5).
--merge_weight	Peso para mesclar Lora no ponto de verificação (padrão: 0,5).

Exemplo 1: Fomer dois modelos Lora

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

Este comando mescla lora1.safetensors e lora2.safetensors , com 70% de contribuição de lora1 .

Exemplo 2: Mesclando Lora em um posto de controle

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

Neste exemplo, o modelo Lora se funde no ponto de verificação com 60% de influência.

Exemplo 3: fusão de dois pontos de verificação

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

Mescla os dois modelos de ponto de verificação em uma mistura de 50-50.

---

Formatos suportados

• .safetensors : Garante armazenamento e manuseio seguros e otimizados dos modelos de IA.

---

Doação intensanceai.art

? Se você achar útil essa ferramenta, considere apoiar o desenvolvimento, consultando nossos planos de preços em Evenenceai.art/PRICE. Ao comprar qualquer plano, você não apenas apoiará futuros projetos de código aberto como esse, mas também desbloqueará muitos recursos exclusivos da IA ​​que se aprimorarão bastante a IA Fluxos de trabalho!

---

Apoiar

Para obter assistência, junte -se ao nosso servidor Discord para se conectar com a comunidade e os desenvolvedores.

---

Contribuindo

Congratulamo -nos com contribuições! Sinta -se à vontade para abrir problemas, enviar solicitações de tração ou sugerir recursos que melhorassem essa ferramenta. Todas as contribuições são muito apreciadas.