Update model.py

model.py

@@ -207,6 +207,8 @@ class Model:
         # condition_img = condition_img.to(self.device)
         # 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)
         # control_image = self.get_control_depth(
         #         image=image,
         #         image_resolution=512,