Spaces:
Paused
Paused
testllama2
Browse files
api/question_answering/qa_model.py
CHANGED
@@ -183,7 +183,7 @@ class QAModel():
|
|
183 |
)
|
184 |
else:
|
185 |
logger.info('using transformers pipeline model')
|
186 |
-
self.llm_model =
|
187 |
model_id=llm_model_id
|
188 |
)
|
189 |
|
|
|
183 |
)
|
184 |
else:
|
185 |
logger.info('using transformers pipeline model')
|
186 |
+
self.llm_model = TransformersPipelineModel(
|
187 |
model_id=llm_model_id
|
188 |
)
|
189 |
|