bhuvanmdev commited on
Commit
f36bd1c
1 Parent(s): f0db4d7

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -1009,7 +1009,7 @@ class MultiViewUNetModel(ModelMixin, ConfigMixin):
1009
  emb = emb + self.camera_embed(camera)
1010
 
1011
  # imagedream variant
1012
- if self.ip_dim > 0 and ip:
1013
  x[(num_frames - 1) :: num_frames, :, :, :] = ip_img # place at [4, 9]
1014
  ip_emb = self.image_embed(ip)
1015
  context = torch.cat((context, ip_emb), 1)
 
1009
  emb = emb + self.camera_embed(camera)
1010
 
1011
  # imagedream variant
1012
+ if self.ip_dim > 0 and ip is not None:
1013
  x[(num_frames - 1) :: num_frames, :, :, :] = ip_img # place at [4, 9]
1014
  ip_emb = self.image_embed(ip)
1015
  context = torch.cat((context, ip_emb), 1)