From fad6b4521fce422f6134e9f41ebd90e4321bbf82 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Mon, 23 Sep 2024 11:11:44 +0300 Subject: [PATCH] fix the case of PAB + GGUF + I2V --- nodes.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/nodes.py b/nodes.py index dea1dbc..129ea8a 100644 --- a/nodes.py +++ b/nodes.py @@ -393,7 +393,10 @@ class DownloadAndLoadCogVideoGGUFModel: transformer = CogVideoXTransformer3DModelFun.from_config(transformer_config) elif "I2V" in model: transformer_config["in_channels"] = 32 - transformer = CogVideoXTransformer3DModel.from_config(transformer_config) + if pab_config is not None: + transformer = CogVideoXTransformer3DModelPAB.from_config(transformer_config) + else: + transformer = CogVideoXTransformer3DModel.from_config(transformer_config) else: transformer_config["in_channels"] = 16 if pab_config is not None: