xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2025-12-24 12:24:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
kijai 2024-10-21 00:50:04 +03:00
parent b33a127dc7
commit 7e8a3e4f2a
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
nodes.py
View File

@ -982,6 +982,9 @@ class CogVideoImageInterpolationEncode:
vae.to(offload_device) vae.to(offload_device)
return ({"samples": final_latents}, ) 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: class ToraEncodeTrajectory:
@classmethod @classmethod