nodes/smthemex-ComfyUI_VisualCloze
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update visualcloze_wrapper.py

Browse Source
This commit is contained in:
smthemex
2025-05-21 16:48:58 +08:00
committed by GitHub
parent 5205bff679
commit a8698c69ed
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
visualcloze_wrapper.py
+3 -3
View File
@@ -296,7 +296,7 @@ class VisualClozeModel:
# when the image is masked, then output it
if mask_position[i - row_start] and is_upsampling:
cropped = output_images[-1].crop(((i - row_start) * ret_w // grid_w, 0, ((i - row_start) + 1) * ret_w // grid_w, ret_h))
cropped.save(f"cropped{i}.png")
#cropped.save(f"cropped{i}.png")
# content clip
for c in content_instruction:
if content_prompt.startswith(c):
@@ -316,7 +316,7 @@ class VisualClozeModel:
ret.append(upsampled)
elif mask_position[i - row_start]:
cropped = output_images[-1].crop(((i - row_start) * ret_w // grid_w, 0, ((i - row_start) + 1) * ret_w // grid_w, ret_h))
cropped.save(f"mask_cropped{i}.png")
#cropped.save(f"mask_cropped{i}.png")
ret.append(cropped)
return ret
return ret