Fix chat template

model.py

@@ -38,9 +38,7 @@ def sample_argmax(logits: torch.Tensor) -> torch.Tensor:
     return torch.argmax(logits, dim=-1)[:, -1]
 
 
-LLAMA_TEMPLATE = """<|begin_of_text|><|start_header_id|>system<|end_header_id|>
-You are a helpful assistant. You give engaging, well-structured answers to user inquiries.<|eot_id|><|start_header_id|>user<|end_header_id|>
-{input}<|eot_id|><|start_header_id|>assistant<|end_header_id|>"""
+LLAMA_TEMPLATE = "<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n\nYou are a helpful assistant. You give engaging, well-structured answers to user inquiries.<|eot_id|><|start_header_id|>user<|end_header_id|>\n\n{input}<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n"
 
 
 class HATCache(Cache):