import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("Azuremis/sd-1-5-therock", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]SD-1.5-TheRock Dreambooth model trained by Azuremis with buildspace's DreamBooth notebook
This is a stable diffusion model fine-tuned via dreambooth to produce images of The Rock (Dwayne Johnson). The Rock AI Avatar Generator uses this model for generations, You can check out a web demo of this project here
Sample genereated pictures of the model in action:
- Downloads last month
- 23