diff --git a/tests/contracts/test_comfyui_tools.py b/tests/contracts/test_comfyui_tools.py index 5d636bd0..a8ee8304 100644 --- a/tests/contracts/test_comfyui_tools.py +++ b/tests/contracts/test_comfyui_tools.py @@ -174,6 +174,20 @@ def test_t2v_workflow_has_templated_nodes(): assert "16" in w # SaveVideo (output) +def test_t2v_workflow_uses_14b_compatible_vae(): + with open(WORKFLOW_DIR / "wan22-t2v-4step.json") as f: + w = json.load(f) + assert w["4"]["inputs"]["vae_name"] == "wan_2.1_vae.safetensors" + + +def test_t2v_metadata_stack_uses_14b_compatible_vae(): + from tools._comfyui.metadata import BUNDLED_MODEL_STACKS + + vae_entry = next(item for item in BUNDLED_MODEL_STACKS["wan22-t2v-4step"] if item["role"] == "vae") + assert vae_entry["name"] == "wan_2.1_vae.safetensors" + assert "Wan_2.1_ComfyUI_repackaged" in vae_entry["download_url"] + + # ------------------------------------------------------------------ # Client unit tests # ------------------------------------------------------------------ diff --git a/tools/_comfyui/metadata.py b/tools/_comfyui/metadata.py index 483beed2..dcca514b 100644 --- a/tools/_comfyui/metadata.py +++ b/tools/_comfyui/metadata.py @@ -86,10 +86,10 @@ BUNDLED_MODEL_STACKS: dict[str, list[dict[str, Any]]] = { }, { "role": "vae", - "name": "wan2.2_vae.safetensors", + "name": "wan_2.1_vae.safetensors", "destination_hint": "ComfyUI/models/vae/", "download_url": ( - "https://huggingface.co/Comfy-Org/Wan_2.2_ComfyUI_Repackaged/" + "https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/" "tree/main/split_files/vae" ), }, diff --git a/tools/_comfyui/workflows/wan22-t2v-4step.json b/tools/_comfyui/workflows/wan22-t2v-4step.json index f5772aca..5016de5e 100644 --- a/tools/_comfyui/workflows/wan22-t2v-4step.json +++ b/tools/_comfyui/workflows/wan22-t2v-4step.json @@ -24,7 +24,7 @@ "4": { "class_type": "VAELoader", "inputs": { - "vae_name": "wan2.2_vae.safetensors" + "vae_name": "wan_2.1_vae.safetensors" } }, "5": {