jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00
Code Issues Activity

Support HunyuanVideo image to video model.

Browse Source
This commit is contained in:
comfyanonymous
2025-03-06 03:07:15 -05:00
parent 0bef826a98
commit 29a70ca101
4 changed files with 132 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
comfy/model_base.py
View File

@@ -900,6 +900,13 @@ class HunyuanVideo(BaseModel):
out['guidance'] = comfy.conds.CONDRegular(torch.FloatTensor([guidance]))
return out
class HunyuanVideoI2V(HunyuanVideo):
def __init__(self, model_config, model_type=ModelType.FLOW, device=None):
super().__init__(model_config, model_type, device=device)
self.concat_keys = ("concat_image", "mask_inverted")
class HunyuanVideoSkyreelsI2V(HunyuanVideo):
def __init__(self, model_config, model_type=ModelType.FLOW, device=None):
super().__init__(model_config, model_type, device=device)