Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -46,21 +46,12 @@ def infer(
|
|
| 46 |
seed = int(randomize_seed_fn(seed, randomize_seed))
|
| 47 |
generator = torch.Generator().manual_seed(seed)
|
| 48 |
|
| 49 |
-
(
|
| 50 |
-
prompt_embeds
|
| 51 |
-
, prompt_neg_embeds
|
| 52 |
-
, pooled_prompt_embeds
|
| 53 |
-
, negative_pooled_prompt_embeds
|
| 54 |
-
) = get_weighted_text_embeddings_sdxl(
|
| 55 |
-
pipe
|
| 56 |
-
, prompt = prompt
|
| 57 |
-
, negative_prompt = negative_prompt
|
| 58 |
-
)
|
| 59 |
image = pipe(
|
| 60 |
-
prompt_embeds
|
| 61 |
-
|
| 62 |
-
|
| 63 |
-
|
| 64 |
#prompt=prompt,
|
| 65 |
#negative_prompt=negative_prompt,
|
| 66 |
width=width,
|
|
|
|
| 46 |
seed = int(randomize_seed_fn(seed, randomize_seed))
|
| 47 |
generator = torch.Generator().manual_seed(seed)
|
| 48 |
|
| 49 |
+
(prompt_embeds, prompt_neg_embeds, pooled_prompt_embeds, negative_pooled_prompt_embeds) = get_weighted_text_embeddings_sdxl(pipe, prompt = prompt, negative_prompt = negative_prompt)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 50 |
image = pipe(
|
| 51 |
+
prompt_embeds = prompt_embeds,
|
| 52 |
+
negative_prompt_embeds = prompt_neg_embeds,
|
| 53 |
+
pooled_prompt_embeds = pooled_prompt_embeds,
|
| 54 |
+
negative_pooled_prompt_embeds = negative_pooled_prompt_embeds,
|
| 55 |
#prompt=prompt,
|
| 56 |
#negative_prompt=negative_prompt,
|
| 57 |
width=width,
|