fix (#711)

modelscope · Apr 16, 2024 · 61785fc · 61785fc
1 parent 3e848b2
commit 61785fc
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/swift/llm/utils/template.py b/swift/llm/utils/template.py
@@ -943,11 +943,10 @@ def get_generate_ids(generate_ids: Tensor,
 
 
 class LLavaYiTemplate(LLavaTemplate):
+    llavayi_query_template = '\n<|im_start|>user\n{{QUERY}}<|im_end|>\n<|im_start|>assistant\n'
 
     def __init__(self):
-        Template.__init__(self, [], [[
-            -200
-        ], '\n<|im_start|>user\n{{QUERY}}<|im_end|>\n<|im_start|>assistant\n'],
+        Template.__init__(self, [], [[-200], self.llavayi_query_template],
                           None, ['<|im_end|>'])