diff --git a/LLM/mlx_language_model.py b/LLM/mlx_language_model.py index ae11b35e7e99e608cea493a5184d18734e2a54eb..82de10214299d4ebe68200256cec73f22974fd02 100644 --- a/LLM/mlx_language_model.py +++ b/LLM/mlx_language_model.py @@ -52,7 +52,7 @@ class MLXLanguageModelHandler(BaseHandler): def warmup(self): logger.info(f"Warming up {self.__class__.__name__}") - dummy_input_text = "Write me a poem about Machine Learning." + dummy_input_text = "Repeat the word 'home'." dummy_chat = [{"role": self.user_role, "content": dummy_input_text}] n_steps = 2