t5 encoder fp8 seems to be working

2026-04-12 13:50:34 -04:00
parent 2818b41004
commit fcf0be38bc
13 changed files with 505 additions and 67 deletions
@@ -105,7 +105,8 @@ def test_models_section_override():
        {
            "models": {
                "wan22_base_repo": "/local/weights/wan22",
-                "wan22_fp8_repo": "/local/weights/wan22-fp8",
+                "wan22_dit_repo": "/local/weights/wan22-dit",
+                "wan22_dit_quant_scheme": "gguf-Q4_K_M",
                "wan22_config_json": "/local/cfg/fp8.json",
                "wan22_model_cls": "wan2.2_moe",
                "musetalk_path": "/local/weights/musetalk",
@@ -113,7 +114,20 @@ def test_models_section_override():
        }
    )
    assert cfg.wan22_base_repo == "/local/weights/wan22"
-    assert cfg.wan22_fp8_repo == "/local/weights/wan22-fp8"
+    assert cfg.wan22_dit_repo == "/local/weights/wan22-dit"
+    assert cfg.wan22_dit_quant_scheme == "gguf-Q4_K_M"
    assert cfg.wan22_config_json == "/local/cfg/fp8.json"
    assert cfg.wan22_model_cls == "wan2.2_moe"
    assert cfg.musetalk_model_path == "/local/weights/musetalk"
+
+
+def test_models_section_backwards_compat_fp8_repo():
+    """Old config key wan22_fp8_repo still works via fallback."""
+    cfg = VideoConfig.from_dict(
+        {
+            "models": {
+                "wan22_fp8_repo": "/local/weights/wan22-fp8",
+            }
+        }
+    )
+    assert cfg.wan22_dit_repo == "/local/weights/wan22-fp8"