Style

style_gen.py

@@ -6,6 +6,7 @@ from typing import Any
 import numpy as np
 import torch
 from numpy.typing import NDArray
+from pyannote.audio import Inference, Model
 from tqdm import tqdm
 
 from config import config
@@ -13,8 +14,6 @@ from style_bert_vits2.logging import logger
 from style_bert_vits2.models.hyper_parameters import HyperParameters
 from style_bert_vits2.utils.stdout_wrapper import SAFE_STDOUT
 
-warnings.filterwarnings("ignore", category=UserWarning)
-from pyannote.audio import Inference, Model
 
 model = Model.from_pretrained("pyannote/wespeaker-voxceleb-resnet34-LM")
 inference = Inference(model, window="whole")