zlei6/ComfyUI
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI.git synced 2026-08-26 02:42:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: copy cropped tensor slices so the uncropped storage can be freed

Browse Source
This commit is contained in:
Terry Jia
2026-08-20 20:47:43 -04:00
parent 4a2058c7c7
commit b71ea70f33
2 changed files with 18 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
comfy_api/latest/_input/video_types.py
View File

@@ -91,11 +91,11 @@ class VideoInput(ABC):
cx, cy, cw, ch = rect cx, cy, cw, ch = rect
return VideoFromComponents( return VideoFromComponents(
VideoComponents( VideoComponents(
images=components.images[:, cy:cy + ch, cx:cx + cw, :], images=components.images[:, cy:cy + ch, cx:cx + cw, :].clone(),
audio=components.audio, audio=components.audio,
frame_rate=components.frame_rate, frame_rate=components.frame_rate,
metadata=components.metadata, metadata=components.metadata,
alpha=components.alpha[:, cy:cy + ch, cx:cx + cw] alpha=components.alpha[:, cy:cy + ch, cx:cx + cw].clone()
if components.alpha is not None if components.alpha is not None
else None, else None,
), ),

16
tests-unit/comfy_api_test/video_types_test.py
View File

@@ -1348,3 +1348,19 @@ def test_cropped_decode_and_save_paths_select_same_pixels(tmp_path):
assert save_column in (2, 3) assert save_column in (2, 3)
assert decode_column == save_column assert decode_column == save_column
def test_as_cropped_components_releases_uncropped_storage():
images = torch.rand(2, 8, 8, 3)
video = VideoFromComponents(
VideoComponents(images=images, frame_rate=Fraction(8))
)
cropped = video.as_cropped(0, 0, 4, 4)
cropped_images = cropped.get_components().images
assert tuple(cropped_images.shape[1:3]) == (4, 4)
assert (
cropped_images.untyped_storage().data_ptr()
!= images.untyped_storage().data_ptr()
)