Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -55,7 +55,7 @@ def error_str(error, title="Error"):
|
|
55 |
return f"""#### {title}
|
56 |
{error}""" if error else ""
|
57 |
|
58 |
-
|
59 |
def inference(prompt, guidance, steps, seed=0, neg_prompt="", disable_auto_prompt_correction=False):
|
60 |
global pipe
|
61 |
|
@@ -109,12 +109,11 @@ def auto_prompt_correction(prompt_ui,neg_prompt_ui,disable_auto_prompt_correctio
|
|
109 |
return prompt, neg_prompt
|
110 |
|
111 |
return prompt,neg_prompt
|
112 |
-
|
|
|
113 |
def txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator):
|
114 |
conditioning, pooled = compel([prompt, neg_prompt])
|
115 |
-
|
116 |
-
pooled=pooled.to("cuda")
|
117 |
-
|
118 |
result = pipe(
|
119 |
prompt_embeds=conditioning[0:1],
|
120 |
pooled_prompt_embeds=pooled[0:1],
|
|
|
55 |
return f"""#### {title}
|
56 |
{error}""" if error else ""
|
57 |
|
58 |
+
|
59 |
def inference(prompt, guidance, steps, seed=0, neg_prompt="", disable_auto_prompt_correction=False):
|
60 |
global pipe
|
61 |
|
|
|
109 |
return prompt, neg_prompt
|
110 |
|
111 |
return prompt,neg_prompt
|
112 |
+
|
113 |
+
@spaces.GPU
|
114 |
def txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator):
|
115 |
conditioning, pooled = compel([prompt, neg_prompt])
|
116 |
+
|
|
|
|
|
117 |
result = pipe(
|
118 |
prompt_embeds=conditioning[0:1],
|
119 |
pooled_prompt_embeds=pooled[0:1],
|