bhuvanmdev
commited on
Commit
•
a7bf0c6
1
Parent(s):
f36bd1c
Update unet/mv_unet.py
Browse files- 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)
|