import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("nagolinc/sd-dune", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]This model was trained based off of https://huggingface.co/runwayml/stable-diffusion-v1-5 for 15000 steps using 2.5k images from https://dune.fandom.com/wiki/Dune_Wiki
Usage:
from diffusers import StableDiffusionPipeline
import torch
pipe=StableDiffusionPipeline.from_pretrained("nagolinc/sd-dune",dtype=torch.float16)
pipe.to("cuda")
image=pipe("shai hulud").images[0]
"bene gesserit"
"dune"
"paul atreides"
"shai hulud"
"taylor swift"
"yoda"
"sandworm"
- Downloads last month
- 4