From f63f22a3ad499d6756448c77076df556dfa84cb6 Mon Sep 17 00:00:00 2001
From: KamioRinn <snowsdream@live.com>
Date: Tue, 6 Aug 2024 17:59:35 +0800
Subject: [PATCH] Turn off onnx warning

---
 GPT_SoVITS/feature_extractor/cnhubert.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/GPT_SoVITS/feature_extractor/cnhubert.py b/GPT_SoVITS/feature_extractor/cnhubert.py
index dc155bdd..3ad90b93 100644
--- a/GPT_SoVITS/feature_extractor/cnhubert.py
+++ b/GPT_SoVITS/feature_extractor/cnhubert.py
@@ -4,8 +4,10 @@ import librosa
 import torch
 import torch.nn.functional as F
 import soundfile as sf
-import logging
+from transformers import logging as tf_logging
+tf_logging.set_verbosity_error()
 
+import logging
 logging.getLogger("numba").setLevel(logging.WARNING)
 
 from transformers import (