Spaces:
Runtime error
Runtime error
Commit ·
31f75a0
1
Parent(s): aa181ce
Update qa_model.py
Browse files- qa_model.py +2 -2
qa_model.py
CHANGED
|
@@ -24,8 +24,8 @@ class QAEnsembleModel_modify(nn.Module):
|
|
| 24 |
AUTH_TOKEN = "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"
|
| 25 |
# model_checkpoint = "letrunglinh/qa_pnc"
|
| 26 |
model_convert = OVModelForQuestionAnswering.from_pretrained(model_name, export= True, use_auth_token= AUTH_TOKEN)
|
| 27 |
-
|
| 28 |
-
|
| 29 |
nlp = pipeline('question-answering', model=model_convert,
|
| 30 |
tokenizer=model_name)
|
| 31 |
self.nlps.append(nlp)
|
|
|
|
| 24 |
AUTH_TOKEN = "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"
|
| 25 |
# model_checkpoint = "letrunglinh/qa_pnc"
|
| 26 |
model_convert = OVModelForQuestionAnswering.from_pretrained(model_name, export= True, use_auth_token= AUTH_TOKEN)
|
| 27 |
+
model_convert.half()
|
| 28 |
+
model_convert.compile()
|
| 29 |
nlp = pipeline('question-answering', model=model_convert,
|
| 30 |
tokenizer=model_name)
|
| 31 |
self.nlps.append(nlp)
|