Fix

4209fd89 · gushiqiao · 8dbe1be6 · 4209fd89
Commit 4209fd89 authored Jul 30, 2025 by gushiqiao
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lightx2v/infer.py lightx2v/infer.py +1 -1

No files found.
--- a/lightx2v/infer.py
+++ b/lightx2v/infer.py
@@ -45,7 +45,7 @@ def main():
        "--model_cls",
        type=str,
        required=True,
-        choices=["wan2.1", "hunyuan", "wan2.1_distill", "wan2.1_causvid", "wan2.1_skyreels_v2_df", "cogvideox", "wan2.1_audio", "wan2.2_moe", "wan2.2"],
+        choices=["wan2.1", "hunyuan", "wan2.1_distill", "wan2.1_causvid", "wan2.1_skyreels_v2_df", "cogvideox", "wan2.1_audio", "wan2.2_moe", "wan2.2_moe_audio", "wan2.2"],
        default="wan2.1",
    )