Skip to content

Commit 5ff4fde

Browse files
authored
Fix LazyCache (Comfy-Org#12344)
1 parent 17e7df4 commit 5ff4fde

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

comfy_extras/nodes_easycache.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,7 @@ def lazycache_predict_noise_wrapper(executor, *args, **kwargs):
108108
easycache: LazyCacheHolder = model_options["transformer_options"]["easycache"]
109109
if easycache.is_past_end_timestep(timestep):
110110
return executor(*args, **kwargs)
111-
x: torch.Tensor = _extract_tensor(args[0], easycache.output_channels)
111+
x: torch.Tensor = args[0][:, :easycache.output_channels]
112112
# prepare next x_prev
113113
next_x_prev = x
114114
input_change = None

0 commit comments

Comments
 (0)