Spaces:

MWilinski
/

bot

Paused

MWilinski commited on Jul 31, 2023

Commit

5ed94e2

1 Parent(s): a7178fa

testllama2

Files changed (1) hide show

api/question_answering/qa_model.py CHANGED Viewed

@@ -183,7 +183,7 @@ class QAModel():
             )
         else:
             logger.info('using transformers pipeline model')
-            self.llm_model = MockLocalBinaryModel(
                 model_id=llm_model_id
             )

             )
         else:
             logger.info('using transformers pipeline model')
+            self.llm_model = TransformersPipelineModel(
                 model_id=llm_model_id
             )