Spaces:
Runtime error
Runtime error
letrunglinh
commited on
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)
|