Update nodes.py
This commit is contained in:
@@ -330,7 +330,7 @@ class HyVideoModelLoader:
|
|||||||
in_channels = sd["img_in.proj.weight"].shape[1]
|
in_channels = sd["img_in.proj.weight"].shape[1]
|
||||||
if in_channels == 16 and "i2v" in model.lower():
|
if in_channels == 16 and "i2v" in model.lower():
|
||||||
i2v_condition_type = "token_replace"
|
i2v_condition_type = "token_replace"
|
||||||
elif in_channels == 16 and not "i2v" in model.lower():
|
elif in_channels == 16 and "custom" in model.lower():
|
||||||
i2v_condition_type = "reference"
|
i2v_condition_type = "reference"
|
||||||
else:
|
else:
|
||||||
i2v_condition_type = "latent_concat"
|
i2v_condition_type = "latent_concat"
|
||||||
|
|||||||
Reference in New Issue
Block a user