working ok

tests/component/test_09_gguf_generate.py

@@ -1,10 +1,10 @@
-"""Quick smoke test: generate a video clip with the GGUF pipeline.
+"""Quick smoke test: generate a video clip with the dense 5B Turbo GGUF pipeline.
 
 Calls Wan22Pipeline.generate_i2v directly (no MuseTalk, no VideoEngine)
 and writes the result to tests/component/_out/phase9_gguf.mp4.
 
 Run:
-    docker compose exec -e DIT_QUANT=gguf-Q4_K_M voice-chat \
+    docker compose exec -e DIT_QUANT=gguf-Q8_0 voice-chat \
         python -m tests.component.test_09_gguf_generate
 """
 from __future__ import annotations
@@ -16,14 +16,9 @@ from tests.component._common import ensure_sample_avatar, get_logger, write_byte
 
 log = get_logger("test_09")
 
-DIT_QUANT = os.environ.get("DIT_QUANT", "gguf-Q4_K_M")
-
-if DIT_QUANT.startswith("gguf-"):
-    CONFIG_JSON = "/app/configs/lightx2v/wan22_i2v_gguf_distill.json"
-    DIT_REPO = "QuantStack/Wan2.2-I2V-A14B-GGUF"
-else:
-    CONFIG_JSON = "/app/configs/lightx2v/wan22_i2v_fp8_distill.json"
-    DIT_REPO = "lightx2v/Wan2.2-Distill-Models"
+DIT_QUANT = os.environ.get("DIT_QUANT", "gguf-Q8_0")
+CONFIG_JSON = "/app/configs/lightx2v/wan22_i2v_gguf_5b_turbo.json"
+DIT_REPO = "hum-ma/Wan2.2-TI2V-5B-Turbo-GGUF"
 
 
 def run():
@@ -38,10 +33,10 @@ def run():
 
     log.info("Building pipeline (quant=%s)...", DIT_QUANT)
     pipe = Wan22Pipeline(
-        base_repo="Wan-AI/Wan2.2-I2V-A14B",
+        base_repo="Wan-AI/Wan2.2-TI2V-5B",
         dit_repo=DIT_REPO,
         config_json=CONFIG_JSON,
-        model_cls="wan2.2_moe_distill",
+        model_cls="wan2.2",
         resolution=480,
         fps=16,
         dit_quant_scheme=DIT_QUANT,