Update app.py
Browse files
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,
|