alfredplpl commited on
Commit
3ac3698
1 Parent(s): a1e7605

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -7
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
- # 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
- 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