Update nodes.py

2025-04-04 03:06:23 +03:00
parent 45e39e758d
commit 78559ef1f4
1 changed files with 2 additions and 1 deletions
@@ -426,7 +426,8 @@ class WanVideoModelLoader:
                new_key = key.replace("model.diffusion_model.", "", 1)
                new_sd[new_key] = value
            sd = new_sd
-
+        if not "patch_embedding.weight" in sd:
+            raise ValueError("Invalid WanVideo model selected")
        dim = sd["patch_embedding.weight"].shape[0]
        in_channels = sd["patch_embedding.weight"].shape[1]
        print("in_channels: ", in_channels)