Spaces:
Runtime error
Runtime error
Damian Stewart
commited on
Commit
·
52c8f3c
1
Parent(s):
6dc9635
fix inference error
Browse files- StableDiffuser.py +1 -1
StableDiffuser.py
CHANGED
|
@@ -209,7 +209,7 @@ class StableDiffuser(torch.nn.Module):
|
|
| 209 |
|
| 210 |
if combined_embeddings is None:
|
| 211 |
assert prompts is not None, "missing prompts or combined_embeddings"
|
| 212 |
-
combined_embeddings =
|
| 213 |
|
| 214 |
width = width or self.native_image_size
|
| 215 |
height = height or self.native_image_size
|
|
|
|
| 209 |
|
| 210 |
if combined_embeddings is None:
|
| 211 |
assert prompts is not None, "missing prompts or combined_embeddings"
|
| 212 |
+
combined_embeddings = self.get_cond_and_uncond_embeddings(prompts, negative_prompts, n_imgs=n_imgs)
|
| 213 |
|
| 214 |
width = width or self.native_image_size
|
| 215 |
height = height or self.native_image_size
|