letrunglinh commited on
Commit
31f75a0
1 Parent(s): aa181ce

Update qa_model.py

Browse files
Files changed (1) hide show
  1. 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
- # model_convert.half()
28
- # model_convert.compile()
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)