From db7b26a8c960380d825baf695388371c4ba49218 Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Thu, 12 Dec 2024 11:15:38 -0800 Subject: [PATCH] remove unused check_model --- .../sentence_transformers/sentence_transformers.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/llama_stack/providers/inline/inference/sentence_transformers/sentence_transformers.py b/llama_stack/providers/inline/inference/sentence_transformers/sentence_transformers.py index bd21698f1..0896b44af 100644 --- a/llama_stack/providers/inline/inference/sentence_transformers/sentence_transformers.py +++ b/llama_stack/providers/inline/inference/sentence_transformers/sentence_transformers.py @@ -41,12 +41,6 @@ class SentenceTransformersInferenceImpl( async def shutdown(self) -> None: pass - def check_model(self, request) -> None: - if request.model != self.config.model: - raise RuntimeError( - f"Model mismatch: {request.model} != {self.config.model}" - ) - async def register_model(self, model: Model) -> None: _ = self._load_sentence_transformer_model(model.provider_resource_id) return model