Spaces:
Running
on
Zero
Running
on
Zero
alfredplpl
commited on
Commit
•
9d5364b
1
Parent(s):
eef0e16
Update app.py
Browse files
app.py
CHANGED
@@ -112,6 +112,8 @@ def auto_prompt_correction(prompt_ui,neg_prompt_ui,disable_auto_prompt_correctio
|
|
112 |
|
113 |
def txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator):
|
114 |
conditioning, pooled = compel([prompt, neg_prompt])
|
|
|
|
|
115 |
|
116 |
result = pipe(
|
117 |
prompt_embeds=conditioning[0:1],
|
|
|
112 |
|
113 |
def txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator):
|
114 |
conditioning, pooled = compel([prompt, neg_prompt])
|
115 |
+
conditioning=conditioning.to("cuda")
|
116 |
+
pooled=pooled.to("cuda")
|
117 |
|
118 |
result = pipe(
|
119 |
prompt_embeds=conditioning[0:1],
|