Hugging Face's logo

stabilityai
/
stable-diffusion-2-base

Text-to-Image
Diffusers
Safetensors
StableDiffusionPipeline
stable-diffusion
Inference Endpoints
Model card Files Community
24
patrickvonplaten commited on
Commit
455a0f4
•
1 Parent(s): 2cfdd58

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -51,7 +51,7 @@ model_id = "stabilityai/stable-diffusion-2-base"
51
 
52
  # Use the Euler scheduler here instead
53
  scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
54
- pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16", torch_dtype=torch.float16)
55
  pipe = pipe.to("cuda")
56
 
57
  prompt = "a photo of an astronaut riding a horse on mars"
 
51
 
52
  # Use the Euler scheduler here instead
53
  scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
54
+ pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)
55
  pipe = pipe.to("cuda")
56
 
57
  prompt = "a photo of an astronaut riding a horse on mars"