Update model.py

model.py

@@ -208,7 +208,7 @@ class Model:
         # condition_img = 2 * (condition_img / 255 - 0.5)
         condition_img = 2 * (image_tensor / 255 - 0.5)
         print(condition_img.shape)
-        condition_img = condition_img.repeat(2, 1, 1, 1)
+        condition_img = condition_img.permute(2,0,1).unsqueeze(0).repeat(2, 1, 1, 1)
         # control_image = self.get_control_depth(
         #         image=image,
         #         image_resolution=512,