Ce nouveau tuto sur Dreambooth stable diffusion 1.5 vous permettra d’ajouter votre tête dans vos générations d’images générées par IA. C’est une mise à jour de l’ancien tuto avec les nouveaux points techniques pour pouvoir générer votre image. Vous pouvez retrouver ce tuto en vidéo.
Colab de Dreambooth stable diffusion 1.5
ll faut se rendre sur https://colab.research.google.com/github/ShivamShrirao/diffusers/blob/main/examples/dreambooth/DreamBooth_Stable_Diffusion.ipynb
Il faut se connecter à Google et suivre toutes les étapes une à une. Je vais expliquer les étapes les plus délicates.
Token HuggingFace
Il faut entrer son token. Il faut aller sur HuggingFace, créer un compte, valider que l’on accepte les conditions générales et copier votre token sur Colab.
Configuration Settings and run
Il faut valider que l’on sauvegarde ses données sur Google drive (pas obligatoire) et peut définir son répertoire où l’on créera son modèle.
Plus bas, il faut configurer son projet en commentant quelques données (celles du chien) et en enlever les # pour celle d’un humain. On utilisera le mot clé toto pour nos images. J’aurai pu choisir n’importe quel autre mot bien entendu.
Préparation de vos images
Il faut envoyer des images de votre visage au format 512×512. Plus il y en a plus le résultat sera meilleur.
Génération des photos
Il faut entrer le mot clé toto dans le prompt afin que notre visage s’affiche.
Exemple : portrait of toto a kratos, cinematic lighting, smooth, focus, cinematic, concept art, highly detailed, artstation, deviantart, behance trending, art by feng zhu, shaddy safadi, noah bradley, tyler edlin, jordan grimmer, darek zabrocki, neil blevins, tuomas korpi, wlop, artgerm
Quelques exemples d’images générées avec ma tête
Réutiliser plus tard le modèle créer
Pour réutiliser le modèle que nous venons de créer, il suffit de récupérer le fichier ckpt qui se trouve dans Mon Drive\stable_diffusion_weights\nomdevotremodel\800\model.ckpt.
Un tuto est disponible sur comment réutiliser un modèle avec Stable Diffusion.