This commit is contained in:
kijai
2025-06-03 08:04:08 +03:00
parent e2f42b5773
commit f4a1157f71
+2 -2
View File
@@ -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}")