diff --git a/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py b/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py index 38d7060..389f05d 100644 --- a/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py +++ b/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py @@ -165,11 +165,6 @@ class HunyuanVideoPipeline(DiffusionPipeline): A scheduler to be used in combination with `unet` to denoise the encoded image latents. """ - #model_cpu_offload_seq = "transformer" - # _optional_components = ["text_encoder_2"] - # _exclude_from_cpu_offload = ["transformer"] - # _callback_tensor_inputs = ["latents", "prompt_embeds", "negative_prompt_embeds"] - def __init__( self, transformer: HYVideoDiffusionTransformer, diff --git a/nodes.py b/nodes.py index eee1abc..049076f 100644 --- a/nodes.py +++ b/nodes.py @@ -15,19 +15,6 @@ from .hyvideo.diffusion.schedulers.scheduling_dpmsolver_multistep import DPMSolv from .hyvideo.diffusion.schedulers.scheduling_sasolver import SASolverScheduler from. hyvideo.diffusion.schedulers.scheduling_unipc_multistep import UniPCMultistepScheduler -# from diffusers.schedulers import ( -# DDIMScheduler, -# PNDMScheduler, -# DPMSolverMultistepScheduler, -# EulerDiscreteScheduler, -# EulerAncestralDiscreteScheduler, -# UniPCMultistepScheduler, -# HeunDiscreteScheduler, -# SASolverScheduler, -# DEISMultistepScheduler, -# LCMScheduler -# ) - scheduler_mapping = { "FlowMatchDiscreteScheduler": FlowMatchDiscreteScheduler, "SDE-DPMSolverMultistepScheduler": DPMSolverMultistepScheduler, diff --git a/pyproject.toml b/pyproject.toml index cabcd99..bc62ad1 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,9 +1,9 @@ [project] -name = "comfyui-hunyuanvideowrapper" +name = "ComfyUI-HunyuanVideoWrapper" description = "ComfyUI diffusers wrapper nodes for [a/HunyuanVideo](https://github.com/Tencent/HunyuanVideo)" -version = "1.0.2" +version = "1.0.3" license = {file = "LICENSE"} -dependencies = ["accelerate >= 1.2.1", "diffusers >= 0.31.0", "transformers >= 4.47.0", "jax >= 0.4.28"] +dependencies = ["accelerate >= 1.2.1", "diffusers >= 0.31.0", "transformers >= 4.47.0", "jax >= 0.4.28", "timm >= 0.6.13"] [project.urls] Repository = "https://github.com/kijai/ComfyUI-HunyuanVideoWrapper"