Update nodes.py

2024-10-21 00:50:04 +03:00
parent b33a127dc7
commit 7e8a3e4f2a
1 changed files with 3 additions and 0 deletions
@@ -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