tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Format import using isort --profile black --gitignore --lai 2 .

Browse Source
This commit is contained in:
litagin02
2024-03-11 12:37:51 +09:00
parent 164b5c4a85
commit dada0af2b6
29 changed files with 50 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
style_bert_vits2/models/attentions.py
View File

@@ -1,6 +1,6 @@
import math
from typing import Any, Optional
import math
import torch
from torch import nn
from torch.nn import functional as F

3
style_bert_vits2/models/commons.py
View File

@@ -3,9 +3,10 @@
コードと完全に一致している保証はない。あくまで参考程度とすること。
"""
from typing import Any, Optional, Union
import torch
from torch.nn import functional as F
from typing import Any, Optional, Union
def init_weights(m: torch.nn.Module, mean: float = 0.0, std: float = 0.01) -> None:

5
style_bert_vits2/models/infer.py
View File

@@ -1,12 +1,11 @@
from typing import Any, cast, Optional, Union
from typing import Any, Optional, Union, cast
import torch
from numpy.typing import NDArray
from style_bert_vits2.constants import Languages
from style_bert_vits2.logging import logger
from style_bert_vits2.models import commons
from style_bert_vits2.models import utils
from style_bert_vits2.models import commons, utils
from style_bert_vits2.models.hyper_parameters import HyperParameters
from style_bert_vits2.models.models import SynthesizerTrn
from style_bert_vits2.models.models_jp_extra import (

5
style_bert_vits2/models/models.py
View File

@@ -7,10 +7,7 @@ from torch.nn import Conv1d, Conv2d, ConvTranspose1d
from torch.nn import functional as F
from torch.nn.utils import remove_weight_norm, spectral_norm, weight_norm
from style_bert_vits2.models import attentions
from style_bert_vits2.models import commons
from style_bert_vits2.models import modules
from style_bert_vits2.models import monotonic_alignment
from style_bert_vits2.models import attentions, commons, modules, monotonic_alignment
from style_bert_vits2.nlp.symbols import NUM_LANGUAGES, NUM_TONES, SYMBOLS

5
style_bert_vits2/models/models_jp_extra.py
View File

@@ -7,10 +7,7 @@ from torch.nn import Conv1d, Conv2d, ConvTranspose1d
from torch.nn import functional as F
from torch.nn.utils import remove_weight_norm, spectral_norm, weight_norm
from style_bert_vits2.models import attentions
from style_bert_vits2.models import commons
from style_bert_vits2.models import modules
from style_bert_vits2.models import monotonic_alignment
from style_bert_vits2.models import attentions, commons, modules, monotonic_alignment
from style_bert_vits2.nlp.symbols import NUM_LANGUAGES, NUM_TONES, SYMBOLS

5
style_bert_vits2/models/monotonic_alignment.py
View File

@@ -3,10 +3,11 @@
コードと完全に一致している保証はない。あくまで参考程度とすること。
"""
from typing import Any
import numba
import torch
from numpy import int32, float32, zeros
from typing import Any
from numpy import float32, int32, zeros
def maximum_path(neg_cent: torch.Tensor, mask: torch.Tensor) -> torch.Tensor:

3
style_bert_vits2/models/utils/__init__.py
View File

@@ -4,7 +4,7 @@ import os
import re
import subprocess
from pathlib import Path
from typing import Any, Optional, Union, TYPE_CHECKING
from typing import TYPE_CHECKING, Any, Optional, Union
import numpy as np
import torch
@@ -15,6 +15,7 @@ from style_bert_vits2.logging import logger
from style_bert_vits2.models.utils import checkpoints # type: ignore
from style_bert_vits2.models.utils import safetensors # type: ignore
if TYPE_CHECKING:
# tensorboard はライブラリとしてインストールされている場合は依存関係に含まれないため、型チェック時のみインポートする
from torch.utils.tensorboard import SummaryWriter