Ceci est un guide sur la façon d’utiliser l’API DreamStudio pour l’IA de génération d’images de diffusion stable en Python.

Stability AI a enfin sorti son IA de génération d’image, Stable Diffusion en Open Source. Parallèlement au code, ils ont également lancé un nouveau produit destiné aux consommateurs appelé DreamStudio

Annonces

DreamStudio est un outil de génération d’images de type OpenAI Dall-e que vous pouvez utiliser dans un navigateur. Ils ont également publié une API pour Dream Studio qui peut être utilisée par les développeurs dans leurs scripts et applications via du code.

Ce guide est destiné aux personnes qui sont programmeurs et codeurs. Si vous cherchez un moyen d’utiliser Stable Diffusion en tant qu’utilisateur final, consultez les guides pour Windows, Mac , Machines sans GPU , etc.

Obtenez une clé API Diffusion DreamStudio stable

Tout d’abord, vous avez besoin de votre propre clé API pour le Stable Diffusion DreamStudio. Voici comment obtenir votre propre clé API Stable Diffusion.

1. Créez un compte ou connectez-vous à DreamStudio
2. Depuis l’icône de votre profil en haut à droite, accédez à votre espace membreAnnonces
3. Cliquez sur Clé API et copiez votre clé API

Conservez-la en lieu sûr et ne la partagez avec personne, car toute personne ayant accès à cette clé peut l’utiliser pour épuiser votre quota de génération.

Utilisation de l’API Stable Diffusion DreamStudio dans un script Python

Si vous préférez appeler l’API dans un script Python sur votre machine locale ou sur un serveur de VM cloud, voici les étapes :

Remarque : crédit de code pour le bloc-notes officiel Stability AI SDK Colab

Assurez-vous que Python 3 ou supérieur est installé sur votre machine et que vous pouvez l’exécuter à partir de la ligne de commande.

python-v

Installer le SDK/module Stable Diffusion Python

 stabilité d' installation de  pip - sdk

Installer libmagic1 

Annonces

Sur Ubuntu

apt - installez libmagic1
 _

Sous Windows

pip install python-libmagic

Créez un nouveau fichier python sur votre IDE préféré et collez le code ci-dessous dans le fichier

Modifiez l’invite dans le code ci-dessus pour obtenir un résultat différent. Assurez-vous également d’ajouter l’API à l’endroit approprié.

importer  io 
importer  os 
importer des  avertissements 
à partir de  PIL  importer une  image 
à partir de  la stabilité_sdk  importer  le client 
importer la  stabilité_sdk.interfaces.gooseai.generation.generation_pb2  en tant que  génération

stabilité_api  =  client . StabilityInference ( 
    clé = 'STABILITY_API_KEY' ,  
    verbeux = True , 
)
 réponses  =  stable_api . générer ( 
    invite = "houston, nous sommes un 'go' pour le lancement!" 
)
 pour le  resp  dans les  réponses : 
    pour l'  artefact  dans le  resp . artefacts : 
        si  artefact . finish_reason  ==  génération . FILTRE :
            avertissements . warn ( 
                "Votre requête a activé les filtres de sécurité de l'API et n'a pas pu être traitée." 
                "Veuillez modifier l'invite et réessayer." ) 
        if  artefact . tapez  ==  génération . ARTEFACT_IMAGE : 
            img  =  Image . ouvrir ( io . BytesIO ( artefact . binaire )) 
            afficher ( img )
---------------------

Exécutez le fichier en le passant à l’interpréteur python

Annonces

python <nom de fichier>

Génération d’image à image à l’aide de l’API DreamStudio

Ci-dessus, nous avons vu comment générer du texte en image à l’aide de l’API DreamStudio.

Nous allons maintenant voir comment utiliser la fonctionnalité d’image à image.

Vous n’avez pas besoin de refaire la configuration et vous pouvez utiliser ce qui est expliqué ci-dessus.

Vous devez passer un paramètre supplémentaire à la fonction de génération d’API

init_image=img

ici img est la variable qui contient l’image que vous souhaitez envoyer à l’API

 

Utilisation de l’API Stable Diffusion DreamStudio dans Jupyter Notebook

1. Ouvrez ce bloc-notes Google Colab officiel.
2. Dans le bloc-notes Google Colab que vous venez d’ouvrir, sélectionnez Runtime dans la barre de menus supérieure, puis cliquez sur Run all
3. Il vous sera demandé d’entrer votre clé API pour DreamStudio. Entrez la clé API que vous avez copiée dans la section ci-dessus et appuyez sur Entrée
4. Une fois terminé, il aurait votre clé API pour que Dream Studio génère une image à l’aide de Stable Diffusion AI.
5. Pour obtenir une image différente, modifiez l’invite dans la dernière cellule et cliquez sur le bouton Exécuter sur le côté gauche de la cellule.

Résumé

Dans cet article, nous avons vu comment utiliser l’API DreamStudio de Stability AI pour générer des images. Nous avons vu comment faire cela en utilisant Python sur une machine locale ainsi que dans Google Colab Notebook.