Spaces:
Running
Running
AbhinavKrishnan36
commited on
Commit
•
f58b81f
1
Parent(s):
ae7a8a2
Update app.py
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ print(response)
|
|
19 |
model_id = "stable-diffusion-v1-5/stable-diffusion-v1-5"
|
20 |
pipe = DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5")
|
21 |
|
22 |
-
pipe.to("
|
23 |
|
24 |
|
25 |
|
|
|
19 |
model_id = "stable-diffusion-v1-5/stable-diffusion-v1-5"
|
20 |
pipe = DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5")
|
21 |
|
22 |
+
pipe.to("cpu")# If you have GPU access; otherwise, use "cpu"
|
23 |
|
24 |
|
25 |
|