Update README.md

README.md

@@ -153,17 +153,45 @@ class Qwen2ForCausalRM(Qwen2ForCausalLM):
 model_path = "TIGER-Lab/AceCodeRM-32B"
 model = Qwen2ForCausalRM.from_pretrained(model_path, device_map="auto")
 tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
+program_correct = """def runningSum(nums):
+    result = []
+    current_sum = 0
+    for num in nums:
+        current_sum += num
+        result.append(current_sum)
+    return result"""
+program_incorrect = """def runningSum(nums):
+    result = []
+    for i in range(len(nums)):
+        if i == 0:
+            result.append(nums[i])
+        else:
+            result.append(nums[i] + nums[i-1])
+    return result"""
 input_chat = [
-                {"role": "user", "content": "Hello, how are you?"},
+        [
+            [
+                {
+                    "content": question,
+                    "role": "user",
+                },
                 {
                     "role": "assistant",
-                    "content": "I'm doing great. How can I help you today?",
+                    "content": program_correct,
                 },
+            ],
+            [
                 {
+                    "content": question,
                     "role": "user",
-                    "content": "I'd like to show off how chat templating works!",
                 },
-            ]
+                {
+                    "role": "assistant",
+                    "content": program_incorrect,
+                },
+            ],
+        ]
+    ]
 input_tokens = tokenizer.apply_chat_template(
         input_chat,
         tokenize=True,