Hugging Face's logo

runwayml
/
stable-diffusion-v1-5

Text-to-Image Diffusers Safetensors StableDiffusionPipeline stable-diffusion stable-diffusion-diffusers Inference Endpoints 5 papers
Model card Files Community
204
patrickvonplaten commited on
Commit
889b629
•
1 Parent(s): 6353453

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -33,7 +33,7 @@ from diffusers import StableDiffusionPipeline
33
  import torch
34
 
35
  model_id = "runwayml/stable-diffusion-v1-5"
36
- pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision="fp16")
37
  pipe = pipe.to("cuda")
38
 
39
  prompt = "a photo of an astronaut riding a horse on mars"
33
  import torch
34
 
35
  model_id = "runwayml/stable-diffusion-v1-5"
36
+ pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
37
  pipe = pipe.to("cuda")
38
 
39
  prompt = "a photo of an astronaut riding a horse on mars"