mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-07-20 21:38:30 +08:00
Copy tensor slice in _flatten_images function
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
This commit is contained in:
parent
1e7e97713c
commit
1f04c50235
@ -77,7 +77,7 @@ def _flatten_images(images):
|
|||||||
continue
|
continue
|
||||||
if image.ndim == 3:
|
if image.ndim == 3:
|
||||||
image = image.unsqueeze(0)
|
image = image.unsqueeze(0)
|
||||||
sources.extend(image[i:i + 1] for i in range(image.shape[0]))
|
sources.extend(image[i:i + 1].clone() for i in range(image.shape[0]))
|
||||||
return sources
|
return sources
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user