working ok

tests/component/test_12_dit_single_step.py

@@ -20,14 +20,9 @@ from tests.component._common import ensure_sample_avatar, get_logger
 
 log = get_logger("test_12")
 
-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():
@@ -42,10 +37,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,