[diffusion] fix: return None instead of raising RuntimeError when no model info found (#21319)
Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
co-authored by
Mick
parent
9238bd08a2
commit
279e7738c5
@@ -349,10 +349,11 @@ def _get_config_info(
|
|||||||
model_id = matched_model_names[0]
|
model_id = matched_model_names[0]
|
||||||
return _CONFIG_REGISTRY.get(model_id)
|
return _CONFIG_REGISTRY.get(model_id)
|
||||||
else:
|
else:
|
||||||
raise RuntimeError(
|
logger.debug(
|
||||||
f"No model info found for model path: {model_path}. "
|
f"No model info found for model path: {model_path}. "
|
||||||
f"Please check the model path or specify the model_id explicitly."
|
f"Please check the model path or specify the model_id explicitly."
|
||||||
)
|
)
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
# --- Part 3: Main Resolver ---
|
# --- Part 3: Main Resolver ---
|
||||||
|
|||||||
Reference in New Issue
Block a user