From 7e8a3e4f2af0ddd499187b09f6d146dcb7518868 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Mon, 21 Oct 2024 00:50:04 +0300
Subject: [PATCH] Update nodes.py

---
 nodes.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/nodes.py b/nodes.py
index 8d27145..aa0e07b 100644
--- 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