diff --git a/src/text_generation/services/nlp/text_generation_completion_service.py b/src/text_generation/services/nlp/text_generation_completion_service.py index 2b3ae4dbc..b0bb4bad0 100644 --- a/src/text_generation/services/nlp/text_generation_completion_service.py +++ b/src/text_generation/services/nlp/text_generation_completion_service.py @@ -177,12 +177,6 @@ class TextGenerationCompletionService(AbstractTextGenerationCompletionService): logger.info(f"Successfully loaded model: {model_id.value}") - def get_current_model_info(self) -> Optional[Dict[str, Any]]: - """Get information about the currently loaded model""" - if self._current_model and self._current_model.is_loaded(): - return self._current_model.get_model_info() - return None - def _process_prompt_with_guidelines_if_applicable(self, user_prompt: str, target_model_id: ModelId): guidelines_config = ( self._use_zero_shot_chain_of_thought, @@ -421,8 +415,8 @@ class TextGenerationCompletionService(AbstractTextGenerationCompletionService): target_model_id = model_id or self._current_model_id or self.default_model_id if (self._current_model_id != target_model_id or - self._current_model is None or - not self._current_model.is_loaded()): + self._current_model is None + ): self.load_model(target_model_id) print(f'Using model: {target_model_id.value}, guidelines: {self.get_current_config()}')