sjrhuschlee commited on
Commit
ab5545b
1 Parent(s): 272d9ce

Update modeling_t5qa.py

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