Refactor: moved model, attentions definitions and inference code to style_bert_vits2/models/

The code has not yet been cleaned up, just moved.
This commit is contained in:
tsukumi
2024-03-06 23:43:25 +00:00
parent a52fda7a88
commit 89825e68d8
10 changed files with 58 additions and 61 deletions

View File

@@ -8,10 +8,6 @@ import torch
from gradio.processing_utils import convert_to_16_bit_wav
import utils
from infer import get_net_g, infer
from models import SynthesizerTrn
from models_jp_extra import SynthesizerTrn as SynthesizerTrnJPExtra
from style_bert_vits2.constants import (
DEFAULT_ASSIST_TEXT_WEIGHT,
DEFAULT_LENGTH,
@@ -23,6 +19,9 @@ from style_bert_vits2.constants import (
DEFAULT_STYLE,
DEFAULT_STYLE_WEIGHT,
)
from style_bert_vits2.models.infer import get_net_g, infer
from style_bert_vits2.models.models import SynthesizerTrn
from style_bert_vits2.models.models_jp_extra import SynthesizerTrn as SynthesizerTrnJPExtra
from style_bert_vits2.logging import logger