sjrhuschlee
/

flan-t5-base-squad2

Question Answering

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

sjrhuschlee commited on Jun 17, 2023

Commit

0fbf44e

•

1 Parent(s): 039280c

Update modeling_t5qa.py

Files changed (1) hide show

modeling_t5qa.py +3 -0

modeling_t5qa.py CHANGED Viewed

@@ -6,6 +6,7 @@ import torch
 from torch import nn
 from torch.nn import CrossEntropyLoss
 from transformers.modeling_outputs import (
     BaseModelOutput,
     Seq2SeqQuestionAnsweringModelOutput,
@@ -199,3 +200,5 @@ class T5ForQuestionAnswering(T5PreTrainedModel):
             encoder_attentions=encoder_outputs.attentions,
         )

 from torch import nn
 from torch.nn import CrossEntropyLoss
+from transformers import AutoModelForQuestionAnswering
 from transformers.modeling_outputs import (
     BaseModelOutput,
     Seq2SeqQuestionAnsweringModelOutput,
             encoder_attentions=encoder_outputs.attentions,
         )
+AutoModelForQuestionAnswering.register(T5Config, T5ForQuestionAnswering)