import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("jzli/DreamShaper-3.3-baked-vae", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
This is DreamShaper 3.3 with vae baked in, so you don't need to add vae when running it
Read more about this model here: https://civitai.com/models/4384/dreamshaper
- Downloads last month
- 11