multi-meeting-QnA

Sleeping

tykiww commited on Jul 31

Commit

be0b5f6

•

1 Parent(s): b817f2f

Update connections/model.py

Files changed (1) hide show

connections/model.py CHANGED Viewed

@@ -23,7 +23,7 @@ class InferencePipeline:
     def infer(self, prompt):
-        outputs = pipeline(
             prompt,
             max_new_tokens=self.conf["model"]["max_new_tokens"],
         )

     def infer(self, prompt):
+        outputs = self.pipeline(
             prompt,
             max_new_tokens=self.conf["model"]["max_new_tokens"],
         )