sjrhuschlee commited on
Commit
ae45ee3
1 Parent(s): d4fe29a

Update modeling_t5qa.py

Browse files
Files changed (1) hide show
  1. modeling_t5qa.py +4 -2
modeling_t5qa.py CHANGED
@@ -200,6 +200,8 @@ class T5ForQuestionAnswering(T5PreTrainedModel):
200
  encoder_attentions=encoder_outputs.attentions,
201
  )
202
 
203
-
204
- AutoModelForQuestionAnswering.register(T5Config, T5ForQuestionAnswering)
 
 
205
 
 
200
  encoder_attentions=encoder_outputs.attentions,
201
  )
202
 
203
+ try:
204
+ AutoModelForQuestionAnswering.register(T5Config, T5ForQuestionAnswering)
205
+ except ValueError:
206
+ pass
207