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

Update unet/mv_unet.py

Browse files
Files changed (1) hide show
  1. unet/mv_unet.py +1 -1
unet/mv_unet.py CHANGED
@@ -985,7 +985,7 @@ class MultiViewUNetModel(ModelMixin, ConfigMixin):
985
  emb = emb + self.camera_embed(camera)
986
 
987
  # imagedream variant
988
- if self.ip_dim > 0 and ip:
989
  x[(num_frames - 1) :: num_frames, :, :, :] = ip_img # place at [4, 9]
990
  ip_emb = self.image_embed(ip)
991
  context = torch.cat((context, ip_emb), 1)
 
985
  emb = emb + self.camera_embed(camera)
986
 
987
  # imagedream variant
988
+ if self.ip_dim > 0 and ip is not None:
989
  x[(num_frames - 1) :: num_frames, :, :, :] = ip_img # place at [4, 9]
990
  ip_emb = self.image_embed(ip)
991
  context = torch.cat((context, ip_emb), 1)