[New Model]: Support GteNewModelForSequenceClassification (#23524)

Signed-off-by: wang.yuqi <noooop@126.com>
2025-08-28 15:36:42 +08:00
parent 186aced5ff
commit 11a7fafaa8
13 changed files with 157 additions and 76 deletions
--- a/tests/conftest.py
+++ b/tests/conftest.py
@ -456,11 +456,10 @@ class HfRunner:
        # output is final logits
        all_inputs = self.get_inputs(prompts)
        outputs = []
+        problem_type = getattr(self.config, "problem_type", "")
+
        for inputs in all_inputs:
            output = self.model(**self.wrap_device(inputs))
-
-            problem_type = getattr(self.config, "problem_type", "")
-
            if problem_type == "regression":
                logits = output.logits[0].tolist()
            elif problem_type == "multi_label_classification":