Update src/tfbench/lm/_hf.py

EYH0602 · Copilot · EYH0602 · commit aace2ad9b5ad · 2025-09-30T13:17:18.000-07:00
Co-authored-by: Copilot &lt;175728472+Copilot@users.noreply.github.com&gt;
diff --git a/src/tfbench/lm/_hf.py b/src/tfbench/lm/_hf.py
@@ -33,8 +33,8 @@ def __init__(self, model_name: str, pure: bool = False):
 
     def _gen(self, prompt: str) -> LMAnswer:
         messages = [
-            {"role": "user", "content": prompt},
             {"role": "system", "content": self.instruction},
+            {"role": "user", "content": prompt},
         ]
         text = self.tokenizer.apply_chat_template(
             messages,

Original file line number	Diff line number	Diff line change
`@@ -33,8 +33,8 @@ def __init__(self, model_name: str, pure: bool = False):`
`33`	`33`
`34`	`34`	`def _gen(self, prompt: str) -> LMAnswer:`
`35`	`35`	`messages = [`
`36`		`- {"role": "user", "content": prompt},`
`37`	`36`	`{"role": "system", "content": self.instruction},`
	`37`	`+ {"role": "user", "content": prompt},`
`38`	`38`	`]`
`39`	`39`	`text = self.tokenizer.apply_chat_template(`
`40`	`40`	`messages,`