mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-10 04:44:57 +08:00
[Bugfix] Fix HfExampleModels.find_hf_info (#12223)
Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
This commit is contained in:
parent
86bfb6dba7
commit
18572e3384
@ -302,6 +302,11 @@ class HfExampleModels:
|
||||
if info.default == model_id:
|
||||
return info
|
||||
|
||||
# Fallback to extras
|
||||
for info in self.hf_models.values():
|
||||
if any(extra == model_id for extra in info.extras.values()):
|
||||
return info
|
||||
|
||||
raise ValueError(f"No example model defined for {model_id}")
|
||||
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user