Fix bug with cosxl edit model.

comfyanonymous · Jun 10, 2024 · a82fae2 · a82fae2
1 parent 8c4a9be
commit a82fae2
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/comfy/conds.py b/comfy/conds.py
@@ -29,7 +29,12 @@ def concat(self, others):
 
 class CONDNoiseShape(CONDRegular):
     def process_cond(self, batch_size, device, area, **kwargs):
-        data = self.cond[:,:,area[2]:area[0] + area[2],area[3]:area[1] + area[3]]
+        data = self.cond
+        if area is not None:
+            dims = len(area) // 2
+            for i in range(dims):
+                data = data.narrow(i + 2, area[i + dims], area[i])
+
         return self._copy_with(comfy.utils.repeat_to_batch_size(data, batch_size).to(device))