qyoo commited on
Commit
456eed0
·
1 Parent(s): 297dfd6
omini_control/__pycache__/condition.cpython-310.pyc CHANGED
Binary files a/omini_control/__pycache__/condition.cpython-310.pyc and b/omini_control/__pycache__/condition.cpython-310.pyc differ
 
omini_control/condition.py CHANGED
@@ -97,8 +97,8 @@ class Condition(object):
97
  cond_tokens = pipe._pack_latents(cond_img, *cond_img.shape)
98
  cond_ids = pipe._prepare_latent_image_ids(
99
  cond_img.shape[0],
100
- cond_img.shape[2],
101
- cond_img.shape[3],
102
  pipe.device,
103
  pipe.dtype,
104
  )
 
97
  cond_tokens = pipe._pack_latents(cond_img, *cond_img.shape)
98
  cond_ids = pipe._prepare_latent_image_ids(
99
  cond_img.shape[0],
100
+ cond_img.shape[2] // 2,
101
+ cond_img.shape[3] // 2,
102
  pipe.device,
103
  pipe.dtype,
104
  )