patrickvonplaten
commited on
Commit
•
51b7853
1
Parent(s):
7c6845f
Update README.md
Browse files
README.md
CHANGED
@@ -35,7 +35,7 @@ import torch
|
|
35 |
|
36 |
model_id = "runwayml/stable-diffusion-v1-5"
|
37 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision="fp16")
|
38 |
-
pipe = pipe.to(
|
39 |
|
40 |
prompt = "a photo of an astronaut riding a horse on mars"
|
41 |
image = pipe(prompt).images[0]
|
|
|
35 |
|
36 |
model_id = "runwayml/stable-diffusion-v1-5"
|
37 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision="fp16")
|
38 |
+
pipe = pipe.to("cuda")
|
39 |
|
40 |
prompt = "a photo of an astronaut riding a horse on mars"
|
41 |
image = pipe(prompt).images[0]
|