hans00 commited on
Commit
625ffa6
1 Parent(s): 0612c39

Update modeling_bert_vits2.py

Browse files
Files changed (1) hide show
  1. 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)