Merge branch 'main' into upgrade-torch-2.3

neuralmagic · May 6, 2024 · 4fa77c0 · 4fa77c0
2 parents 0266449 + 1bad1fb
commit 4fa77c0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sparseml/transformers/finetune/data/gsm8k.py b/src/sparseml/transformers/finetune/data/gsm8k.py
@@ -28,7 +28,7 @@ class GSM8KDataset(TextGenerationDataset):
     :param tokenizer: tokenizer to use on dataset
     """
 
-    GSM_TEMPLATE = "Question: {question}.\nAnswer:"
+    GSM_TEMPLATE = "Question: {question}\nAnswer:"
 
     def __init__(self, data_args, split, tokenizer):
         data_args = deepcopy(data_args)