Fix: support infer 2.2 models (#244)

* Fix: support infer 2.2 models * Fix imports * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-12-21 22:59:54 +09:00
parent 7ebc1aa28a
commit 98f5917ac9
30 changed files with 133754 additions and 79 deletions
--- a/models.py
+++ b/models.py
@@ -14,8 +14,6 @@ from torch.nn.utils import weight_norm, remove_weight_norm, spectral_norm
 from commons import init_weights, get_padding
 from text import symbols, num_tones, num_languages

-from vector_quantize_pytorch import VectorQuantize
-

 class DurationDiscriminator(nn.Module):  # vits2
    def __init__(