From ee522e693f446d7d810e3a8316eea97075b87c04 Mon Sep 17 00:00:00 2001 From: llyycchhee Date: Tue, 9 Jun 2026 03:07:10 +0000 Subject: [PATCH] fix(model_registration):add flexible type --- xinference/core/worker.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/xinference/core/worker.py b/xinference/core/worker.py index baa0d0ebb9..6d9f1ff058 100644 --- a/xinference/core/worker.py +++ b/xinference/core/worker.py @@ -1516,6 +1516,12 @@ async def get_model_registration(self, model_type: str, model_name: str) -> Any: for f in get_user_defined_reranks(): if f.model_name == model_name: return self._prefer_model_hub(f) + elif model_type == "flexible": + from ..model.flexible import get_flexible_models + + for f in get_flexible_models(): + if f.model_name == model_name: + return f return None @log_async(logger=logger)