"container/vscode:/vscode.git/clone" did not exist on "abd4b5d9a09bd81a0d940100d09b460a700092f3"
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,9 +622,14 @@ def get_config( ...@@ -622,9 +622,14 @@ def get_config(
# Architecture mapping for models without explicit architectures field # Architecture mapping for models without explicit architectures field
if not config.architectures: if not config.architectures:
if config.model_type not in MODEL_MAPPING_NAMES: if config.model_type not in MODEL_MAPPING_NAMES:
raise ValueError(f"Cannot find architecture name for {config.model_type}") logger.warning(
model_type = MODEL_MAPPING_NAMES[config.model_type] "Model config does not have a top-level 'architectures' field: "
config.update({"architectures": [model_type]}) "expecting `hf_overrides={'architectures': ['...']}` to be passed "
"in engine args."
)
else:
model_type = MODEL_MAPPING_NAMES[config.model_type]
config.update({"architectures": [model_type]})
# ModelOpt 0.31.0 and after saves the quantization config in the model # ModelOpt 0.31.0 and after saves the quantization config in the model
# config file. # config file.
......
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