import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("nexuslrf/nd-diffusion", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]Normal-Depth Diffusion Model
This is an unofficial export of the original normal-depth diffusion model (https://github.com/modelscope/normal-depth-diffusion) to the standard diffusers model format.
Can be direcetly loaded via diffusers' StableDiffusionPipeline
All the credit to the original author of normal-depth diffusion model (https://github.com/modelscope/normal-depth-diffusion)
- Downloads last month
- 8