[DeepSparse Evaluation API] Add model property to DeepSparseLM (#…

…1606) * Update lm_evaluation_harness.py * Update lm_evaluation_harness.py
neuralmagic · Feb 20, 2024 · a284a0a · a284a0a
1 parent b0a066e
commit a284a0a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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