Spaces:
Runtime error
Runtime error
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,
|