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

fix the case of PAB + GGUF + I2V

Browse Source
This commit is contained in:
kijai 2024-09-23 11:11:44 +03:00
parent 7f3a768d95
commit fad6b4521f
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
nodes.py
View File

@ -393,6 +393,9 @@ class DownloadAndLoadCogVideoGGUFModel:
transformer = CogVideoXTransformer3DModelFun.from_config(transformer_config) transformer = CogVideoXTransformer3DModelFun.from_config(transformer_config)
elif "I2V" in model: elif "I2V" in model:
transformer_config["in_channels"] = 32 transformer_config["in_channels"] = 32
if pab_config is not None:
transformer = CogVideoXTransformer3DModelPAB.from_config(transformer_config)
else:
transformer = CogVideoXTransformer3DModel.from_config(transformer_config) transformer = CogVideoXTransformer3DModel.from_config(transformer_config)
else: else:
transformer_config["in_channels"] = 16 transformer_config["in_channels"] = 16