Unverified Commit a8d2e326 authored by Roger Wang's avatar Roger Wang Committed by GitHub
Browse files

[Bugfix][CI] Fix config resolving logic with remote models (#27610)

parent 53a56e65
......@@ -622,7 +622,12 @@ def get_config(
# Architecture mapping for models without explicit architectures field
if not config.architectures:
if config.model_type not in MODEL_MAPPING_NAMES:
raise ValueError(f"Cannot find architecture name for {config.model_type}")
logger.warning(
"Model config does not have a top-level 'architectures' field: "
"expecting `hf_overrides={'architectures': ['...']}` to be passed "
"in engine args."
)
else:
model_type = MODEL_MAPPING_NAMES[config.model_type]
config.update({"architectures": [model_type]})
......
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