diff --git a/src/deepsparse/evaluation/integrations/lm_evaluation_harness.py b/src/deepsparse/evaluation/integrations/lm_evaluation_harness.py index 69934af37a..38d9fdf4bd 100644 --- a/src/deepsparse/evaluation/integrations/lm_evaluation_harness.py +++ b/src/deepsparse/evaluation/integrations/lm_evaluation_harness.py @@ -150,6 +150,10 @@ def max_length(self) -> int: def max_gen_toks(self) -> int: return self._max_gen_toks + @property + def model(self) -> Pipeline: + return self.pipeline + def loglikelihood(self, requests) -> List[Tuple[float, bool]]: """ Copied directly from