Picture Yourself with Stable Diffusion

Key Takeaways

hey guys welcome to another tutorial about stable diffusion today I want to show you an easy way how to create your own custom stable diffusion model using just a few random photos of yourself and the free online tool to train your model first I will show you what to do in four easy steps and then I will guide you through each step in detail first you need to take some photos of yourself or any other person you want to train be sure that the photos have a good contrast and lighting no hard Shadows the whole face should be visible there shouldn't be any other people on the photo and it's good to use different background angles and different facial expressions I would recommend taking 15 photos more isn't necessary next you need to resize your photos exactly 512 by 512 pixels because that's the resolution of the stable diffusion training model then you need to give them a unique name and that's very important because it will be the keyword for your prompts later in stable diffusion in my case I chose verolab because I'm going to train a model of Vero who is the band leader of our music project maybe you can also visit our music channel on YouTube where you can find lots of videos made with stable diffusion I will leave a link down below so I renamed all images to eurolab and a sequence number starting from 1 up to 15. once you've finished that just upload the 15 images to your Google Drive then we can open dream pools to train our model and once the training is finished you just need to download your model from your Google Drive import it into your stable diffusion app and then you can start creating images or videos with your very personal model okay enough for Theory now to the Practical part I've already prepared 15 images of Vero it's good if the images already have a rather Square format because we will need to resize them to 512 by 512 in the next step you can do that manually but there's a great free website called burme.net where you can resize a whole bunch of images in one step I will leave a link down below just upload your images to burme then set the size to 512 by 512 and download them as a zip file next let's rename the images to virolab 1 to 15. and then let's open our Google Drive it's drive.google.com if you haven't got the Google Drive just create a free Gmail address go to Google and login into your browser and then your Google Drive will be ready let's create a new folder called virolab then double click to open it and drag the 15 resized photos to your drive okay now everything is set up for the training we are going to use a Google collab file called the last pen fast to input I will leave a link down below so the training will not be done locally on your machine but on a Google server it might look a bit intimidating at first but it's in fact fairly easy first click on copy to drive to make a local copy of The Notebook and then let's connect to your Google drive to this notebook just click on the icon from Google collab import drive you will be asked for permission to connect your Google Drive just allow it and we are done with the first step when you click on the folder icon left hand side you can see that the Google Drive is connected and you can see the images we just uploaded before next click on the little icon left to dependencies and wait a little while until they are installed next we need to choose our model version which has stay with 1.5 and then we need to select the base model and train our virolab images into this model we could just use the standard stable diffusion 1.5 basic model but I want to use a different model which I find more realistic and in order to that let's head over to civitai.com which is a great free website for lots and lots of different stable diffusion models just take a closer look on this side and choose a model that you like I'm going with an epic realistic model which is great for creating photorealistic images just note that it's a safe tensor file and not a simple checkpoint we will need this information later now right click on the download button and copy the link address then let's go back to our Dream pools notebook and paste the link into the line model link and because it's a safe tensor file also hit the save tensor checkbox and run the model download script this might take a few minutes I'll be back when it's done okay next let's create a session I just call it virolab and click on the icon to run the script next we need to tell the system on which images it should be trained so click on those three dots on the image folder and hit copy path then paste the path into the image folder line and then run the script we can skip the next two cells because we won't need them so let's head right over to our training cell the training steps are set to 1500 which is perfectly fine for us because you need about 100 training steps for each image just if you have more or less images to train you need to change this value you can leave the rest of the settings as they are because I found that it produces rather good results so just click on the icon to start the training now this might take up to 20 minutes so I'll be back when it's finished all right we are done with the training so let's disconnect from The Notebook and delete the runtime because we won't need it anymore then let's go back to our Google Drive and find our model that just has been created on the top level of your drive you can find a folder fast dream Booth inside that folder is a sessions folder and inside that there's another folder where you can find your model just look for the file with the ckpt ending and download this file to your local machine now the last thing we need to do is importing the model into our stable diffusion app I'm using automatic 1111 which I think is one of the best apps available I will leave a link down below how to install it basically you can use any stable diffusion app you like for automatic 11 11 you need to move the model file into the stable diffusion web UI folder models stable diffusion and basically that's all that's needed now let's open Automatic 1111 and select our custom bundle you might need to hit the refresh button if the model isn't visible quite a way okay then once the model is loaded let's enter a simple test prompt in order to see if our model works you just need to take sure to use the keyword verolab which is the trigger word for our custom model then hit the generate button and see what happens looks quite okay for the first try now you can play with more sophisticated prompts and get some real great results well I think that's all for today I will leave a very detailed description of the whole process down below and also some additional information about the tools being used in that video and some more general information about stable diffusion I hope this video was helpful for you and if you have any further questions don't hesitate asking so bye for now and see you in the next one