Update nodes.py

2025-12-24 12:24:27 +08:00 · 2024-10-21 00:50:04 +03:00 · 2024-10-21 00:50:04 +03:00 · 7e8a3e4f2a
commit 7e8a3e4f2a
parent b33a127dc7
1 changed files with 3 additions and 0 deletions
--- a/nodes.py
+++ b/nodes.py
@ -983,6 +983,9 @@ class CogVideoImageInterpolationEncode:
        
        return ({"samples": final_latents}, )
    
+from .tora.traj_utils import process_traj, scale_traj_list_to_256
+from torchvision.utils import flow_to_image
+
 class ToraEncodeTrajectory:
    @classmethod
    def INPUT_TYPES(s):