Update modeling_bert_vits2.py
Browse files- modeling_bert_vits2.py +1 -2
modeling_bert_vits2.py
CHANGED
|
@@ -1380,7 +1380,6 @@ class BertVits2PreTrainedModel(PreTrainedModel):
|
|
| 1380 |
models.
|
| 1381 |
"""
|
| 1382 |
|
| 1383 |
-
config_class = BertVits2Config
|
| 1384 |
base_model_prefix = "vits"
|
| 1385 |
main_input_name = "input_ids"
|
| 1386 |
supports_gradient_checkpointing = True
|
|
@@ -1494,7 +1493,7 @@ class BertVits2Model(BertVits2PreTrainedModel):
|
|
| 1494 |
return self.text_encoder
|
| 1495 |
|
| 1496 |
@add_start_docstrings_to_model_forward(BERT_VITS2_INPUTS_DOCSTRING)
|
| 1497 |
-
@replace_return_docstrings(output_type=BertVits2ModelOutput
|
| 1498 |
def forward(
|
| 1499 |
self,
|
| 1500 |
input_ids: Optional[torch.Tensor] = None,
|
|
|
|
| 1380 |
models.
|
| 1381 |
"""
|
| 1382 |
|
|
|
|
| 1383 |
base_model_prefix = "vits"
|
| 1384 |
main_input_name = "input_ids"
|
| 1385 |
supports_gradient_checkpointing = True
|
|
|
|
| 1493 |
return self.text_encoder
|
| 1494 |
|
| 1495 |
@add_start_docstrings_to_model_forward(BERT_VITS2_INPUTS_DOCSTRING)
|
| 1496 |
+
@replace_return_docstrings(output_type=BertVits2ModelOutput)
|
| 1497 |
def forward(
|
| 1498 |
self,
|
| 1499 |
input_ids: Optional[torch.Tensor] = None,
|