justinhl
/

hybrid-qa1

Question Answering

text-generation

text-generation-inference

Model card Files Files and versions

justinhl commited on May 14, 2024

Commit

ceb5437

·

verified ·

1 Parent(s): 46202d7

Update hybrid_pipe.py

Files changed (1) hide show

hybrid_pipe.py +1 -1

hybrid_pipe.py CHANGED Viewed

@@ -101,7 +101,7 @@ class HybridQAModel(nn.Module, PyTorchModelHubMixin):
         model.model_extractive = AutoModelForQuestionAnswering.from_pretrained(save_directory, trust_remote_code=True)
         model.tokenizer_extractive = AutoTokenizer.from_pretrained(save_directory, trust_remote_code=True)
-        model.model_generative = AutoModelForSeq2SeqLM.from_pretrained(save_directory, trust_remote_code=True)
         model.tokenizer_generative = AutoTokenizer.from_pretrained(save_directory, trust_remote_code=True)
         return model

         model.model_extractive = AutoModelForQuestionAnswering.from_pretrained(save_directory, trust_remote_code=True)
         model.tokenizer_extractive = AutoTokenizer.from_pretrained(save_directory, trust_remote_code=True)
+        model.model_generative = AutoModelForCausalLM.from_pretrained(save_directory, trust_remote_code=True)
         model.tokenizer_generative = AutoTokenizer.from_pretrained(save_directory, trust_remote_code=True)
         return model