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

Update nodes.py

Browse Source
This commit is contained in:
kijai 2024-10-21 00:49:07 +03:00
parent 4d437dae66
commit b33a127dc7
1 changed files with 0 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
nodes.py
View File

@ -983,41 +983,6 @@ class CogVideoImageInterpolationEncode:
return ({"samples": final_latents}, )
class ToraEncodeTrajectory:
@classmethod
def INPUT_TYPES(s):
return {"required": {
"pipeline": ("COGVIDEOPIPE",),
"width": ("INT", {"default": 720, "min": 128, "max": 2048, "step": 8}),
"height": ("INT", {"default": 480, "min": 128, "max": 2048, "step": 8}),
"num_frames": ("INT", {"default": 49, "min": 16, "max": 1024, "step": 1}),
},
}
RETURN_TYPES = ("TORATRAJLIST",)
RETURN_NAMES = ("tora_traj_list",)
FUNCTION = "encode"
CATEGORY = "CogVideoWrapper"
def encode(self, pipeline, width, height, num_frames):
device = mm.get_torch_device()
offload_device = mm.unet_offload_device()
generator = torch.Generator(device=device).manual_seed(0)
transformer = pipeline["pipe"].transformer
vae = pipeline["pipe"].vae
vae.enable_slicing()
canvas_width, canvas_height = 256, 256
traj_list = PROVIDED_TRAJS["infinity"]
traj_list_range_256 = scale_traj_list_to_256(traj_list, canvas_width, canvas_height)
return (traj_list_range_256, )
from .tora.traj_utils import process_traj, scale_traj_list_to_256, PROVIDED_TRAJS
from torchvision.utils import flow_to_image
class ToraEncodeTrajectory:
@classmethod
def INPUT_TYPES(s):