optimize: config

This commit is contained in:
源文雨 2023-08-27 00:35:39 +08:00
parent 80bf0cb8c8
commit 8c36dc2cc7
5 changed files with 144 additions and 146 deletions

4
app.py
View File

@ -9,7 +9,7 @@ import logging
from fairseq import checkpoint_utils from fairseq import checkpoint_utils
from vc_infer_pipeline import VC from vc_infer_pipeline import VC
import traceback import traceback
from config import Config from config import defaultconfig as config
from lib.infer_pack.models import ( from lib.infer_pack.models import (
SynthesizerTrnMs256NSFsid, SynthesizerTrnMs256NSFsid,
SynthesizerTrnMs256NSFsid_nono, SynthesizerTrnMs256NSFsid_nono,
@ -26,8 +26,6 @@ logging.getLogger("matplotlib").setLevel(logging.WARNING)
i18n = I18nAuto() i18n = I18nAuto()
i18n.print() i18n.print()
config = Config()
weight_root = "weights" weight_root = "weights"
weight_uvr5_root = "uvr5_weights" weight_uvr5_root = "uvr5_weights"
index_root = "logs" index_root = "logs"

View File

@ -198,3 +198,5 @@ class Config:
except: except:
pass pass
return x_pad, x_query, x_center, x_max return x_pad, x_query, x_center, x_max
defaultconfig = Config()

View File

@ -22,7 +22,7 @@ import faiss
import ffmpeg import ffmpeg
import gradio as gr import gradio as gr
import soundfile as sf import soundfile as sf
from config import Config from config import defaultconfig as config
import fairseq import fairseq
from i18n import I18nAuto from i18n import I18nAuto
from lib.infer_pack.models import ( from lib.infer_pack.models import (
@ -53,7 +53,6 @@ warnings.filterwarnings("ignore")
torch.manual_seed(114514) torch.manual_seed(114514)
config = Config()
if config.dml == True: if config.dml == True:
def forward_dml(ctx, x, scale): def forward_dml(ctx, x, scale):

View File

@ -13,11 +13,10 @@ import scipy.signal as signal
now_dir = os.getcwd() now_dir = os.getcwd()
sys.path.append(now_dir) sys.path.append(now_dir)
from config import Config from config import defaultconfig as config
from multiprocessing import Manager as M from multiprocessing import Manager as M
mm = M() mm = M()
config = Config()
if config.dml == True: if config.dml == True:
def forward_dml(ctx, x, scale): def forward_dml(ctx, x, scale):