Merge Diffusion Tool

Outil de diffusion fusion

L'outil de diffusion de fusion est une solution open source développée par Enhanceai.art , fournissant des moyens sans couture de mélanger les modèles LORA, d'intégrer LORA dans des points de contrôle et de fusionner les points de contrôle de diffusion stable. Améliorez vos flux de travail AI avec cet outil de fusion puissant, conçu pour prendre en charge une large gamme de modèles de diffusion comme Flux Dev , Flux Schnell , stable Diffusion 1.5 , SD2 , SD3 et SDXL .

Découvrez des outils d'IA plus avancés sur Enhanceai.art.

Caractéristiques

• Fusionner deux modèles LORA avec des rapports de mélange réglables.
• Intégrez LORA dans un point de contrôle pour les performances améliorées du modèle.
• Fusionner deux points de contrôle à l'aide de ratios de mélange personnalisés.
• Soutien complet au format .safetensors , assurant une manipulation efficace et sûre.

Avantages clés:

• Rationalisez votre processus de fusion du modèle AI.
• Construit avec flexibilité pour répondre à divers formats de modèles d'IA.
• Complètement open-source avec le soutien communautaire via Discord.

---

Installation

Pour configurer l'outil, suivez ces étapes:

1. Clone le référentiel:

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

2. Installez les dépendances requises:

   pip install -r requirements.txt

Dépendances:

• Python 3.8+
• Pytorch pour les opérations d'apprentissage en profondeur.
• Bibliothèque Safettensers pour la gestion des fichiers du modèle.

---

Usage

L' outil de diffusion de fusion fournit trois opérations principales:

1. Fusionner deux modèles LORA
2. Fusionner Lora dans un point de contrôle
3. Fusionner deux points de contrôle

Arguments en ligne de commande:

Argument	Description
--operation	Sélectionnez l'opération: merge_loras , merge_lora_checkpoint , merge_checkpoints
--model1	Chemin vers le premier fichier de modèle.
--model2	Chemin vers le deuxième fichier de modèle (facultatif pour LORA dans le point de contrôle).
--output	Chemin de sortie pour le modèle fusionné.
--alpha	Ratio de mélange pour les modèles de fusion (par défaut: 0,5).
--merge_weight	Poids pour fusionner LORA dans le point de contrôle (par défaut: 0,5).

Exemple 1: fusion de deux modèles LORA

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

Cette commande fusionne lora1.safetensors et lora2.safetensors , avec une contribution de 70% de lora1 .

Exemple 2: fusion de Lora dans un point de contrôle

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

Dans cet exemple, le modèle LORA se fusionne dans le point de contrôle avec une influence de 60%.

Exemple 3: fusion de deux points de contrôle

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

Merge les deux modèles de point de contrôle dans un mélange 50-50.

---

Formats pris en charge

• .safetensors : assure un stockage et une manipulation sûrs et optimisés des modèles d'IA.

---

Donation Enhanceai.art

? Si vous trouvez cet outil utile, veuillez envisager de soutenir le développement en consultant nos plans de prix sur Enhanceai.art/pricing. En achetant n'importe quel plan, non seulement vous soutiendrez de futurs projets open source comme celui-ci, mais vous débloquerez également de nombreuses fonctionnalités d'IA exclusives qui vous amélioreront considérablement les flux de travail de l'IA!

---

Soutien

Pour obtenir de l'aide, rejoignez notre serveur Discord pour vous connecter avec la communauté et les développeurs.

---

Contributif

Nous accueillons les contributions! N'hésitez pas à ouvrir des problèmes, à soumettre des demandes de traction ou à suggérer des fonctionnalités qui amélioreraient cet outil. Toutes les contributions sont très appréciées.