Update hybrid_pipe.py
Browse files- hybrid_pipe.py +1 -1
hybrid_pipe.py
CHANGED
@@ -101,7 +101,7 @@ class HybridQAModel(nn.Module, PyTorchModelHubMixin):
|
|
101 |
|
102 |
model.model_extractive = AutoModelForQuestionAnswering.from_pretrained(save_directory, trust_remote_code=True)
|
103 |
model.tokenizer_extractive = AutoTokenizer.from_pretrained(save_directory, trust_remote_code=True)
|
104 |
-
model.model_generative =
|
105 |
model.tokenizer_generative = AutoTokenizer.from_pretrained(save_directory, trust_remote_code=True)
|
106 |
|
107 |
return model
|
|
|
101 |
|
102 |
model.model_extractive = AutoModelForQuestionAnswering.from_pretrained(save_directory, trust_remote_code=True)
|
103 |
model.tokenizer_extractive = AutoTokenizer.from_pretrained(save_directory, trust_remote_code=True)
|
104 |
+
model.model_generative = AutoModelForCausalLM.from_pretrained(save_directory, trust_remote_code=True)
|
105 |
model.tokenizer_generative = AutoTokenizer.from_pretrained(save_directory, trust_remote_code=True)
|
106 |
|
107 |
return model
|