wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-15 03:27:24 +08:00
Code Issues Actions 22 Packages Projects Releases Wiki Activity

Clamp out-of-range negative batch_index to start of batch

Browse Source
This commit is contained in:
ozbayb 2026-05-12 16:48:06 -06:00
parent a927795108
commit 73bdfaf639
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy_extras/nodes_images.py
View File

@ -147,7 +147,7 @@ class ImageFromBatch(IO.ComfyNode):
s_in = image
if batch_index < 0:
batch_index += s_in.shape[0]
batch_index = min(s_in.shape[0] - 1, batch_index)
batch_index = max(0, min(s_in.shape[0] - 1, batch_index))
length = min(s_in.shape[0] - batch_index, length)
s = s_in[batch_index:batch_index + length].clone()
return IO.NodeOutput(s)

2
nodes.py
View File

@ -1234,7 +1234,7 @@ class LatentFromBatch:
s_in = samples["samples"]
if batch_index < 0:
batch_index += s_in.shape[0]
batch_index = min(s_in.shape[0] - 1, batch_index)
batch_index = max(0, min(s_in.shape[0] - 1, batch_index))
length = min(s_in.shape[0] - batch_index, length)
s["samples"] = s_in[batch_index:batch_index + length].clone()
if "noise_mask" in samples: