ford442 commited on
Commit
6beca96
·
verified ·
1 Parent(s): 1918ae9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -13
app.py CHANGED
@@ -191,17 +191,17 @@ def generate_30(
191
  negative_prompt_embeds,
192
  pooled_prompt_embeds,
193
  negative_pooled_prompt_embeds
194
- ) = pipe.encode_prompt(prompt)
195
  del pipe.text_encoder, pipe.text_encoder_2
196
  flush()
197
  options = {
198
  # "prompt": [prompt],
199
  # "negative_prompt": [negative_prompt],
200
  # "negative_prompt_2": [neg_prompt_2],
201
- "prompt_embeds": [prompt_embeds],
202
- "negative_prompt_embeds": [negative_prompt_embeds],
203
- "pooled_prompt_embeds": [pooled_prompt_embeds],
204
- "negative_pooled_prompt_embeds": [negative_pooled_prompt_embeds],
205
  "width": width,
206
  "height": height,
207
  "guidance_scale": guidance_scale,
@@ -252,10 +252,10 @@ def generate_60(
252
  # "prompt": [prompt],
253
  # "negative_prompt": [negative_prompt],
254
  # "negative_prompt_2": [neg_prompt_2],
255
- "prompt_embeds": [prompt_embeds],
256
- "negative_prompt_embeds": [negative_prompt_embeds],
257
- "pooled_prompt_embeds": [pooled_prompt_embeds],
258
- "negative_pooled_prompt_embeds": [negative_pooled_prompt_embeds],
259
  "width": width,
260
  "height": height,
261
  "guidance_scale": guidance_scale,
@@ -306,10 +306,10 @@ def generate_90(
306
  # "prompt": [prompt],
307
  # "negative_prompt": [negative_prompt],
308
  # "negative_prompt_2": [neg_prompt_2],
309
- "prompt_embeds": [prompt_embeds],
310
- "negative_prompt_embeds": [negative_prompt_embeds],
311
- "pooled_prompt_embeds": [pooled_prompt_embeds],
312
- "negative_pooled_prompt_embeds": [negative_pooled_prompt_embeds],
313
  "width": width,
314
  "height": height,
315
  "guidance_scale": guidance_scale,
 
191
  negative_prompt_embeds,
192
  pooled_prompt_embeds,
193
  negative_pooled_prompt_embeds
194
+ ) = pipe.encode_prompt(prompt,negative_prompt)
195
  del pipe.text_encoder, pipe.text_encoder_2
196
  flush()
197
  options = {
198
  # "prompt": [prompt],
199
  # "negative_prompt": [negative_prompt],
200
  # "negative_prompt_2": [neg_prompt_2],
201
+ "prompt_embeds": prompt_embeds,
202
+ "negative_prompt_embeds": negative_prompt_embeds,
203
+ "pooled_prompt_embeds": pooled_prompt_embeds,
204
+ "negative_pooled_prompt_embeds": negative_pooled_prompt_embeds,
205
  "width": width,
206
  "height": height,
207
  "guidance_scale": guidance_scale,
 
252
  # "prompt": [prompt],
253
  # "negative_prompt": [negative_prompt],
254
  # "negative_prompt_2": [neg_prompt_2],
255
+ "prompt_embeds": prompt_embeds,
256
+ "negative_prompt_embeds": negative_prompt_embeds,
257
+ "pooled_prompt_embeds": pooled_prompt_embeds,
258
+ "negative_pooled_prompt_embeds": negative_pooled_prompt_embeds,
259
  "width": width,
260
  "height": height,
261
  "guidance_scale": guidance_scale,
 
306
  # "prompt": [prompt],
307
  # "negative_prompt": [negative_prompt],
308
  # "negative_prompt_2": [neg_prompt_2],
309
+ "prompt_embeds": prompt_embeds,
310
+ "negative_prompt_embeds": negative_prompt_embeds,
311
+ "pooled_prompt_embeds": pooled_prompt_embeds,
312
+ "negative_pooled_prompt_embeds": negative_pooled_prompt_embeds,
313
  "width": width,
314
  "height": height,
315
  "guidance_scale": guidance_scale,