Update context_windows.py

This commit is contained in:
kijai 2025-11-22 15:15:13 +02:00
parent 928fd4d13f
commit ac93376ef8

View File

@ -564,6 +564,7 @@ def shift_window_to_end(window: list[int], num_frames: int):
# 2) add end_delta to each val to slide windows to end
window[i] = window[i] + end_delta
# https://github.com/Kosinkadink/ComfyUI-AnimateDiff-Evolved/blob/90fb1331201a4b29488089e4fbffc0d82cc6d0a9/animatediff/sample_settings.py#L465
def apply_freenoise(noise: torch.Tensor, context_length: int, context_overlap: int, seed: int):
logging.info(f"Context windows: Applying FreeNoise")
@ -585,4 +586,4 @@ def apply_freenoise(noise: torch.Tensor, context_length: int, context_overlap: i
list_idx = torch.tensor(list(range(start_idx,start_idx+delta)), device=torch.device("cpu"), dtype=torch.long)
list_idx = list_idx[torch.randperm(delta, generator=generator)]
noise[:, :, place_idx:place_idx + delta] = noise[:, :, list_idx]
return noise
return noise