diff --git a/src/abstractions/model.py b/src/abstractions/model.py index 861a67c..ad5619f 100644 --- 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")):