sjrhuschlee
commited on
Commit
•
ab5545b
1
Parent(s):
272d9ce
Update modeling_t5qa.py
Browse files- 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
|