Spaces:
Running
on
Zero
Running
on
Zero
alfredplpl
commited on
Commit
•
3ac3698
1
Parent(s):
a1e7605
Update app.py
Browse files
app.py
CHANGED
@@ -115,17 +115,16 @@ def txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator):
|
|
115 |
conditioning, pooled = compel([prompt, neg_prompt])
|
116 |
|
117 |
result = pipe(
|
118 |
-
|
119 |
-
|
120 |
-
|
121 |
-
|
122 |
-
prompt=prompt,
|
123 |
-
negative=neg_prompt,
|
124 |
num_inference_steps = int(steps),
|
125 |
guidance_scale = guidance,
|
126 |
width = width,
|
127 |
height = height,
|
128 |
-
generator = generator
|
|
|
129 |
|
130 |
return result.images[0]
|
131 |
|
|
|
115 |
conditioning, pooled = compel([prompt, neg_prompt])
|
116 |
|
117 |
result = pipe(
|
118 |
+
prompt_embeds=conditioning[0:1],
|
119 |
+
pooled_prompt_embeds=pooled[0:1],
|
120 |
+
negative_prompt_embeds=conditioning[1:2],
|
121 |
+
negative_pooled_prompt_embeds=pooled[1:2],
|
|
|
|
|
122 |
num_inference_steps = int(steps),
|
123 |
guidance_scale = guidance,
|
124 |
width = width,
|
125 |
height = height,
|
126 |
+
generator = generator
|
127 |
+
)
|
128 |
|
129 |
return result.images[0]
|
130 |
|