diff --git a/src/deepsparse/evaluation/integrations/lm_evaluation_harness.py b/src/deepsparse/evaluation/integrations/lm_evaluation_harness.py index 845fe062ac..6e0711cb3c 100644 --- a/src/deepsparse/evaluation/integrations/lm_evaluation_harness.py +++ b/src/deepsparse/evaluation/integrations/lm_evaluation_harness.py @@ -155,6 +155,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