Copy tensor slice in _flatten_images function

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
This commit is contained in:
Silver 2026-07-13 19:20:51 +02:00 committed by GitHub
parent 1e7e97713c
commit 1f04c50235
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -77,7 +77,7 @@ def _flatten_images(images):
continue
if image.ndim == 3:
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