Tutoriel Stable Diffusion : Création d'une API sur une instance VM GCP

Comprendre Stable Diffusion : Un outil IA révolutionnaire

Stable Diffusion est un modèle avancé d'apprentissage profond introduit en 2022, conçu pour transformer des descriptions textuelles en images époustouflantes. Ce modèle de pointe de conversion texte-image a gagné en immense popularité grâce à sa capacité à générer des visuels détaillés à partir de simples entrées textuelles. Au-delà de la génération d'images, Stable Diffusion sert à plusieurs fins telles que l'inpainting, l'outpainting et la génération de traductions d'image à image guidées par des invites textuelles. Stability.ai a joué un rôle central dans le développement du modèle, s'assurant qu'il reste à la pointe de la technologie créative IA.

Comment créer un compte Google Cloud Platform (GCP)

Si vous souhaitez exploiter la puissance de Stable Diffusion, la première étape consiste à créer un compte GCP. Suivez ces étapes :

Visitez la page GCP Free Tier.
Configurez un compte de facturation, car l'utilisation du GPU nécessite des informations de facturation.
Envisagez de définir un budget avec des alertes pour gérer les coûts efficacement.

Demande d'accès GPU dans votre projet GCP

Après avoir créé votre compte GCP, vous devrez activer l'API Compute Engine :

Naviguez vers la page APIs & Services.
Recherchez l'API Compute Engine et cliquez sur Activer.
Demandez la permission de créer des machines virtuelles avec GPU en allant sur la page Quotas. Filtrez pour les GPU et demandez une augmentation de 0 à 1, en spécifiant une raison comme l'utilisation d'un modèle ML nécessitant un GPU.

L'approbation peut prendre quelques jours, donc la patience est de mise.

Création d'une instance de machine virtuelle (VM)

Ensuite, vous devrez créer une instance de VM :

Allez sur Page des instances GCP et cliquez sur Créer une instance.
Nommez votre instance (par exemple, stable-diffusion-instance) et sélectionnez votre région souhaitée.
Dans la configuration de la machine, sélectionnez GPU. Le A100 offre des performances de pointe, mais le T4 est plus économique et efficace pour la plupart des usages.
Type de machine : Sélectionnez n1-standard-4 avec au moins 15 Go de mémoire.
Pour le disque de démarrage, changez-le pour une VM d'apprentissage profond basée sur Debian 10, et augmentez la taille du disque si nécessaire.
Sous Pare-feu, cochez Autoriser le trafic HTTP et Autoriser le trafic HTTPS.
Dans Réseautage, ajoutez une balise réseau (par exemple, stable-diffusion-tag).

Un des avantages des instances GCP est que vous ne payez que lorsqu'elles sont actives.

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

Pour garantir que votre instance soit accessible depuis Internet, créez une règle de pare-feu :

Allez dans la section Règles de pare-feu et cliquez sur Créer une règle de pare-feu.
Nommez-la stable-diffusion-rule et sélectionnez Tags pour ajouter votre balise précédente.
Dans Plages d'IP source, ajoutez 0.0.0.0/0 pour permettre l'accès depuis n'importe quelle IP.
Dans Protocoles et ports, spécifiez tcp:5000 et cliquez sur Créer.

Accéder et configurer votre instance de calcul

Pour accéder à l'instance, utilisez SSH via la console. Lors de la première connexion, vous devrez installer le pilote Nvidia en tapant Y. Notez que vous devrez peut-être réinstaller les pilotes après le redémarrage de votre VM.

Mise en place de Stable Diffusion

Une fois l'instance configurée, vous clonerez deux référentiels et installerez Cog, comme indiqué sur sa page GitHub. Après la configuration, construisez votre image Docker qui nécessitera le téléchargement de poids depuis Hugging Face :

Créez un compte sur Hugging Face.
Générez un jeton d'authentification depuis votre page de paramètres.
Utilisez votre jeton pour télécharger les poids du modèle nécessaires.

Une fois tout configuré, vous pouvez tester si le modèle fonctionne correctement.

Tester votre configuration

Après avoir exécuté le modèle, un fichier de sortie nommé output-1.png devrait apparaître. Téléchargez-le via le terminal pour vérifier le bon fonctionnement. De plus, vous pouvez récupérer l'IP externe de votre instance de calcul pour accéder au service sur Internet.

Décodage des réponses API

Lors de la test d'API, notez que les réponses sont encodées en base64. Pour décoder, utilisez des outils en ligne tels que Code Beautify pour plus de commodité.

Explorer les fonctionnalités supplémentaires

Pour ceux intéressés par la génération de vidéos, le modèle Deforum est une excellente option pour créer de courtes vidéos en utilisant des méthodes similaires. Assurez-vous d'arrêter les conteneurs en cours lors de l'essai de nouveaux modèles !

Conclusion

Merci d'avoir suivi ce tutoriel ! Si vous l'avez trouvé utile, consultez d'autres ressources sur notre page de tutoriels et continuez d'explorer les capacités de Stable Diffusion.

Ce guide vise à vous permettre d'exploiter le potentiel de Stable Diffusion et GCP pour vos projets créatifs. Bonne informatique !