Update src/condition.py

src/condition.py

@@ -96,8 +96,8 @@ class Condition(object):
         cond_tokens = pipe._pack_latents(cond_img, *cond_img.shape)
         cond_ids = pipe._prepare_latent_image_ids(
             cond_img.shape[0],
-            cond_img.shape[2],
-            cond_img.shape[3],
+            cond_img.shape[2]//2,
+            cond_img.shape[3]//2,
             pipe.device,
             pipe.dtype,
         )