Spaces:
Sleeping
Sleeping
yuvaranianandhan24
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -73,7 +73,7 @@ def txt2img(text, style="realistic"):
|
|
73 |
# Use the Euler scheduler here instead
|
74 |
scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
|
75 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)
|
76 |
-
pipe = pipe.to("cuda")
|
77 |
image = pipe(prompt = text, guidance_scale = 7.5).images[0]
|
78 |
return image
|
79 |
|
|
|
73 |
# Use the Euler scheduler here instead
|
74 |
scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
|
75 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)
|
76 |
+
#pipe = pipe.to("cuda")
|
77 |
image = pipe(prompt = text, guidance_scale = 7.5).images[0]
|
78 |
return image
|
79 |
|