diff --git a/nodes.py b/nodes.py index 7cc2c3d..d17fa4a 100644 --- a/nodes.py +++ b/nodes.py @@ -2476,8 +2476,8 @@ class WanVideoSampler: from .ATI.motion_patch import patch_motion topk = transformer_options.get("ati_topk", 2) temperature = transformer_options.get("ati_temperature", 220.0) - ati_start_percent = transformer_options.get("ati_start_percentage", 0.0) - ati_end_percent = transformer_options.get("ati_end_percentage", 1.0) + ati_start_percent = transformer_options.get("ati_start_percent", 0.0) + ati_end_percent = transformer_options.get("ati_end_percent", 1.0) image_cond_ati = patch_motion(ATI_tracks.to(image_cond.device, image_cond.dtype), image_cond, topk=topk, temperature=temperature) log.info(f"ATI tracks shape: {ATI_tracks.shape}")