Créer une API Stable Diffusion sur GCP : Un guide étape par étape

Qu'est-ce que la diffusion stable ?

La diffusion stable est un modèle d'apprentissage profond de pointe lancé en 2022 qui se spécialise dans la génération d'images à partir de descriptions textuelles. Il permet aux utilisateurs de créer des images de haute qualité à partir de descriptions textuelles. La flexibilité du modèle s'étend à diverses applications, y compris l'inpainting, l'outpainting et la génération de traductions d'images en les guidant avec des invites textuelles. Avec le soutien de Stability.ai, la diffusion stable a rapidement gagné en popularité parmi les développeurs et les artistes.

Comment créer un compte GCP

Pour utiliser efficacement la diffusion stable, vous devez créer un compte Google Cloud Platform (GCP) si vous ne l'avez pas déjà fait. Suivez ces étapes :

1. Visitez GCP Free Tier et inscrivez-vous.
2. Créez un compte de facturation, car les fonctionnalités GPU nécessitent des comptes activés pour la facturation.
3. Envisagez d'établir un budget avec des alertes pour éviter des dépenses imprévues, en particulier si l'utilisation du GPU peut s'accumuler rapidement.

Demande d'accès GPU

Une fois votre compte configuré, suivez ces étapes pour demander un accès GPU :

1. Activez l'API Compute Engine en naviguant vers la page APIs & Services et en la recherchant.
2. Cliquez sur l'API puis sur Activer.
3. Dans la section Quota, filtrez pour les GPU (toutes les régions) et demandez une augmentation de 0 à 1, en expliquant que vous utiliserez un modèle d'apprentissage automatique nécessitant un support GPU.
4. Notez que l'approbation de votre demande peut prendre jusqu'à deux jours.

Création d'une instance de machine virtuelle

Après l'approbation de votre accès GPU, il est temps de créer une instance VM :

1. Allez sur GCP Compute Instances et cliquez sur Créer une instance.
2. Nommez votre instance ; par ex. "stable-diffusion-instance".
3. Sélectionnez la région avec soin, car la disponibilité des GPU peut varier.
4. Choisissez un type de machine ; le NVIDIA T4 est une option solide et économique.
5. Pour le disque de démarrage, réglez-le sur Deep Learning Linux avec au moins 50 Go de stockage.
6. Activez les paramètres de pare-feu requis en cochant à la fois Autoriser le trafic HTTP et Autoriser le trafic HTTPS.

Création d'une règle de pare-feu

Avant d'accéder à votre instance, configurez une règle de pare-feu :

1. Allez sur GCP Firewall Rules et sélectionnez Créer une règle de pare-feu.
2. Nommez-la stable-diffusion-rule et ajoutez votre balise précédente dans la section des cibles.
3. Définissez les plages d'IP sources sur 0.0.0.0/0 et autorisez tcp:5000 via les protocoles et ports.

Accéder à votre instance de calcul

Le moyen le plus simple d'accéder à votre nouvelle instance VM est via SSH depuis votre console GCP. Retournez à votre liste d'instances et cliquez sur votre instance.

Configuration de la diffusion stable

Lors de votre première connexion, il vous sera demandé d'installer les pilotes NVIDIA :

1. Suivez l'invite en tapant 'Y' pour oui afin d'installer les pilotes.
2. Veuillez noter que vous devrez réinstaller ces pilotes après tout arrêt/démarrage de l'instance.
3. Clonez les dépôts nécessaires requis pour la diffusion stable.
4. Assurez-vous que Docker est installé.
5. Enfin, construisez votre image Docker et téléchargez les poids du modèle depuis Hugging Face après avoir créé un compte et généré un jeton d'authentification.

Tester votre modèle

Une fois les installations terminées et le modèle en cours d'exécution, vous recevrez un fichier de sortie (généralement output-1.png). Vous pouvez le télécharger facilement via des commandes terminal.

Pour trouver votre répertoire de travail actuel, tapez pwd et notez le chemin pour les étapes suivantes.

Test de l'API

Les réponses de votre modèle seront encodées au format base64. Utilisez des outils comme convertisseur base64 en image pour décoder et prévisualiser vos images.

Exploration de Deforum

Si vous êtes intéressé par la création de vidéos, envisagez de consulter Deforum, un modèle particulièrement conçu pour créer de courtes vidéos.

Conclusion

Merci d'avoir suivi ce guide sur la configuration de la diffusion stable sur GCP ! Pour plus de tutoriels, n'hésitez pas à visiter notre page de tutoriels et à explorer davantage.

Ces instructions ont été compilées à partir de diverses sources pour fournir des conseils complets sur l'atteinte de vos objectifs d'apprentissage automatique avec la diffusion stable.