fix(evaluation): remove requirement for alpaca template

PKU-Alignment · Dec 21, 2024 · 4d909d9 · 4d909d9
1 parent fb7e2d4
commit 4d909d9
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/src/abstractions/model.py b/src/abstractions/model.py
@@ -1098,11 +1098,6 @@ def evaluate(
             )
 
     def __evaluate_fast(self, logprobs=True) -> np.ndarray:
-        if self.template_type != "alpaca":
-            raise NotImplementedError(
-                "Fast evaluation is only supported for models using alpaca template."
-            )
-
         if not os.path.exists(f"{root}/output"):
             os.mkdir(f"{root}/output")
         if not os.path.exists(os.path.join(root, "output", "evaluation_results")):