accuracy enhancement for baichuan2-13B for npu (#16868)

Co-authored-by: cy <chenyang08056032@163.com>
2026-01-26 16:14:35 +08:00
parent 6756cf18c6
commit 2734b23481
5 changed files with 241 additions and 75 deletions
@@ -22,6 +22,7 @@ class GSM8KAscendMixin(ABC):
        "ascend",
        "--disable-cuda-graph",
    ]
+    gsm8k_num_shots = 5

    @classmethod
    def setUpClass(cls):
@@ -52,7 +53,7 @@ class GSM8KAscendMixin(ABC):

    def test_gsm8k(self):
        args = SimpleNamespace(
-            num_shots=5,
+            num_shots=self.gsm8k_num_shots,
            data_path=None,
            num_questions=200,
            max_new_tokens=512,