Damian Stewart commited on
Commit
52c8f3c
1 Parent(s): 6dc9635

fix inference error

Browse files
Files changed (1) hide show
  1. 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 = diffuser.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
 
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