Stable Diffusion Finetuning and Deployment
Recommended tutorials before starting to work with stable diffusion finetuning and deployment jobs.
1. Create and upload datasets
Firstly we need to upload your images to our platform before finetuning stable diffusion model using these images. Navigate through the Stochastic Platform in the Datasets side tab and select Add dataset. Next, add your dataset name, dataset description, choose Image as type and browse your images folder. Then click Upload to create an images dataset.
The number of images should be higher than 5 and lower than 30 and the original resolution should be higher than 512x512.
2. Create finetuning job
Navigate through the Jobs side tab and select Create job. You can select Stable Diffusion 1 or Stable Diffusion 2 as the initialized pretrained model for your finetuning job. Stable Diffusion 2 is chosen in this example.
Next, choose your uploaded images dataset and put your Class name. Class name of the images dataset subject. Is recommended using generic classes such as man, woman, or child (if the subject is a person) or cat or dog (if the subject is a pet).
Finally, select number of saved checkpoints and click Start to start finetune Stable Diffusion on your dataset.
The finetuning job usually takes about 2 hours for Stable Diffusion 2 and about 1.2 hours for Stable Diffusion 1.
3. Deploy finetuned model
After your finetuning job succeeded. Navigate through the Jobs side tab and expand to see your saved checkpoints. Click Save to choose the checkpoint which will be deployed.
Navigate through the Deploy side tab and click Create deployments to deploy your saved models. Next, put your deployment name and select your saved model then click Deploy to deploy it.
Wait a bit until for deployment done then click to it to checkout how to use your deployment to generate images. We also provided example code to execute the API example for each programing language.
Note that your prompt should be start with sks prefix and include the class name you used in the finetuning phase. Below is the example with the prompt a photo of sks handsome football player raising the world cup championship trophy with img_height = 768 and img_width = 768.
