Spaces:
nebula
/
Running on Zero

Menyu commited on
Commit
b5902d0
·
verified ·
1 Parent(s): 896044c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -46,11 +46,12 @@ def infer(
46
  seed = int(randomize_seed_fn(seed, randomize_seed))
47
  generator = torch.Generator().manual_seed(seed)
48
  compel = Compel(tokenizer=pipe.tokenizer, text_encoder=pipe.text_encoder)
49
- conditioning = compel.build_conditioning_tensor(prompt)
50
 
51
  image = pipe(
52
  #prompt=prompt,
53
  prompt_embeds=conditioning,
 
54
  negative_prompt=negative_prompt,
55
  width=width,
56
  height=height,
 
46
  seed = int(randomize_seed_fn(seed, randomize_seed))
47
  generator = torch.Generator().manual_seed(seed)
48
  compel = Compel(tokenizer=pipe.tokenizer, text_encoder=pipe.text_encoder)
49
+ conditioning, pooled = compel.build_conditioning_tensor(prompt)
50
 
51
  image = pipe(
52
  #prompt=prompt,
53
  prompt_embeds=conditioning,
54
+ pooled_prompt_embeds=pooled,
55
  negative_prompt=negative_prompt,
56
  width=width,
57
  height=height,