Fix model selection for I2V generation by checking capabilities before scoring

parent 0a0e2084
......@@ -3947,6 +3947,16 @@ def select_best_model(gen_type, models, vram_gb=24, prefer_quality=True, return_
# Check model capabilities
caps = detect_model_type(info)
# Skip models that don't support the required generation type
if gen_type_str == "t2v" and not caps["t2v"]:
continue
elif gen_type_str == "i2v" and not caps["i2v"]:
continue
elif gen_type_str == "t2i" and not caps["t2i"]:
continue
elif gen_type_str == "i2i" and not caps["i2i"]:
continue
# Score the model
score = 0
reasons = []
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment