Skip to content

Commit c3e385a

Browse files
DarkLight1337rasmith
authored andcommitted
[Bugfix] Fix HfExampleModels.find_hf_info (vllm-project#12223)
Signed-off-by: DarkLight1337 <[email protected]>
1 parent 458adaa commit c3e385a

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/models/registry.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -302,6 +302,11 @@ def find_hf_info(self, model_id: str) -> _HfExamplesInfo:
302302
if info.default == model_id:
303303
return info
304304

305+
# Fallback to extras
306+
for info in self.hf_models.values():
307+
if any(extra == model_id for extra in info.extras.values()):
308+
return info
309+
305310
raise ValueError(f"No example model defined for {model_id}")
306311

307312

0 commit comments

Comments
 (0)