Update modeling_bert_vits2.py

modeling_bert_vits2.py

@@ -33,6 +33,7 @@ from transformers.modeling_outputs import (
 from transformers.models.bert.modeling_bert import BertModel
 from transformers.modeling_utils import PreTrainedModel
 from transformers.utils import logging
+from configuration_bert_vits2 import BertVits2Config
 
 logger = logging.get_logger(__name__)
 
@@ -1379,7 +1380,7 @@ class BertVits2PreTrainedModel(PreTrainedModel):
     An abstract class to handle weights initialization and a simple interface for downloading and loading pretrained
     models.
     """
-
+    config_class = BertVits2Config
     base_model_prefix = "vits"
     main_input_name = "input_ids"
     supports_gradient_checkpointing = True
@@ -1404,10 +1405,6 @@ class BertVits2PreTrainedModel(PreTrainedModel):
                 module.weight.data[module.padding_idx].zero_()
 
 
-# drop config_class for BertVits2PreTrainedModel
-del BertVits2PreTrainedModel.config_class
-
-
 class BertVits2Model(BertVits2PreTrainedModel):
     def __init__(self, config):
         super().__init__(config)