zlei6/OpenMontage
1
0
Fork 0
mirror of https://github.com/calesthio/OpenMontage.git synced 2026-08-24 17:10:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(comfyui): pair wan22 t2v workflow with 14b-compatible vae

Browse Source
This commit is contained in:
Your Name
2026-07-04 19:45:47 +05:30
parent f4b8b90a24
commit 2beda014be
3 changed files with 17 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/contracts/test_comfyui_tools.py
View File

@@ -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
# ------------------------------------------------------------------