From 18572e3384a6f55a7589dd81e1f3f70f7dd73e3a Mon Sep 17 00:00:00 2001 From: Cyrus Leung Date: Mon, 20 Jan 2025 23:35:36 +0800 Subject: [PATCH] [Bugfix] Fix `HfExampleModels.find_hf_info` (#12223) Signed-off-by: DarkLight1337 --- tests/models/registry.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/models/registry.py b/tests/models/registry.py index 23227ea6b971..e99dbd16c47b 100644 --- a/tests/models/registry.py +++ b/tests/models/registry.py @@ -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}")