Update nodes.py

2025-12-09 21:04:23 +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
@ -982,6 +982,9 @@ class CogVideoImageInterpolationEncode:
            vae.to(offload_device)
        
        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