Update modeling_enc_t5.py

modeling_enc_t5.py

@@ -4,7 +4,7 @@ from typing import Any, Dict, List, Optional
 import torch
 from torch import nn
 from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss
-from transformers import T5TokenizerFast
+from transformers import T5TokenizerFast, T5Config
 from transformers.modeling_outputs import SequenceClassifierOutput, TokenClassifierOutput
 from transformers.models.t5.modeling_t5 import T5Config, T5PreTrainedModel, T5Stack
 from transformers.utils.model_parallel_utils import assert_device_map, get_device_map
@@ -224,5 +224,6 @@ class EncT5ForTokenClassification(T5PreTrainedModel):
         )
 
 
+T5Config.register_for_auto_class()
 EncT5Tokenizer.register_for_auto_class("AutoTokenizer")
 EncT5ForTokenClassification.register_for_auto_class("AutoModelForTokenClassification")