Update modeling_bert_vits2.py
Browse files- modeling_bert_vits2.py +4 -0
modeling_bert_vits2.py
CHANGED
@@ -1404,6 +1404,10 @@ class BertVits2PreTrainedModel(PreTrainedModel):
|
|
1404 |
module.weight.data[module.padding_idx].zero_()
|
1405 |
|
1406 |
|
|
|
|
|
|
|
|
|
1407 |
class BertVits2Model(BertVits2PreTrainedModel):
|
1408 |
def __init__(self, config):
|
1409 |
super().__init__(config)
|
|
|
1404 |
module.weight.data[module.padding_idx].zero_()
|
1405 |
|
1406 |
|
1407 |
+
# drop config_class for BertVits2PreTrainedModel
|
1408 |
+
del BertVits2PreTrainedModel.config_class
|
1409 |
+
|
1410 |
+
|
1411 |
class BertVits2Model(BertVits2PreTrainedModel):
|
1412 |
def __init__(self, config):
|
1413 |
super().__init__(config)
|