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