fix chat template bug

Blaizzy · Aug 16, 2024 · bef76d8 · bef76d8
1 parent fe842f3
commit bef76d8
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/mlx_vlm/prompt_utils.py b/mlx_vlm/prompt_utils.py
@@ -33,9 +33,7 @@ def get_message_json(model_name, prompt):
 def apply_chat_template(processor, config, prompt):
     message = get_message_json(config["model_type"], prompt)
 
-    if "chat_template" in processor.__dict__.keys() and hasattr(
-        processor, "default_chat_template"
-    ):
+    if "chat_template" in processor.__dict__.keys():
         return processor.apply_chat_template(
             [message],
             tokenize=False,