How to use sourceoftruthdata/sot_autotrain_dreambooth_v1 with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("sourceoftruthdata/sot_autotrain_dreambooth_v1", dtype=torch.bfloat16, device_map="cuda") prompt = "a hand drawn painting in the style of picasso with geometric shapes" image = pipe(prompt).images[0]
Text encoder was not trained.This is the model that feeds the Google Colab Notebook.The model is a simplified version of the DreamBooth model.
Type in a prompt, get an image.
The user is responsible for utilizing this application.
Base model