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