[Bugfix] Fix HfExampleModels.find_hf_info (vllm-project#12223)

Signed-off-by: DarkLight1337 <[email protected]>

Author: DarkLight1337

tests/models/registry.py

@@ -302,6 +302,11 @@ def find_hf_info(self, model_id: str) -> _HfExamplesInfo:
             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}")