Spaces:
Build error
Build error
Commit
·
89859a3
1
Parent(s):
6ae7ae6
Update app.py
Browse files
app.py
CHANGED
@@ -32,7 +32,7 @@ def img_Generation(text):
|
|
32 |
# Use the Euler scheduler here instead
|
33 |
scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
|
34 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16", torch_dtype=torch.float16)
|
35 |
-
|
36 |
image = pipe(text, num_inference_steps = 150).images[0]
|
37 |
#image.save("img_1.png")
|
38 |
|
|
|
32 |
# Use the Euler scheduler here instead
|
33 |
scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
|
34 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16", torch_dtype=torch.float16)
|
35 |
+
pipe = pipe.to("cuda")
|
36 |
image = pipe(text, num_inference_steps = 150).images[0]
|
37 |
#image.save("img_1.png")
|
38 |
|