Cool_Tools/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of synced 2025-01-31 20:05:20 +01:00
Code Releases Wiki Activity

Update gui.py (#338)

Browse Source
This commit is contained in:
JackEllie 2023-05-25 08:27:40 +08:00 committed by GitHub
parent e0813eb282
commit 039e7afb85
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
gui.py
View File

@ -14,9 +14,9 @@ import os, sys, traceback
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 Config as MyConfig
is_half = Config().is_half is_half = MyConfig().is_half
import PySimpleGUI as sg import PySimpleGUI as sg
import sounddevice as sd import sounddevice as sd
import noisereduce as nr import noisereduce as nr
@ -48,6 +48,7 @@ class RVC:
初始化 初始化
""" """
try: try:
self.config = MyConfig()
self.f0_up_key = key self.f0_up_key = key
self.time_step = 160 / 16000 * 1000 self.time_step = 160 / 16000 * 1000
self.f0_min = 50 self.f0_min = 50
@ -80,17 +81,18 @@ class RVC:
cpt["config"][-3] = cpt["weight"]["emb_g.weight"].shape[0] # n_spk cpt["config"][-3] = cpt["weight"]["emb_g.weight"].shape[0] # n_spk
self.if_f0 = cpt.get("f0", 1) self.if_f0 = cpt.get("f0", 1)
self.version = cpt.get("version", "v1") self.version = cpt.get("version", "v1")
if version == "v1":
if if_f0 == 1: if self.version == "v1":
if self.if_f0 == 1:
self.net_g = SynthesizerTrnMs256NSFsid( self.net_g = SynthesizerTrnMs256NSFsid(
*cpt["config"], is_half=config.is_half *cpt["config"], is_half=self.config.is_half
) )
else: else:
self.net_g = SynthesizerTrnMs256NSFsid_nono(*cpt["config"]) self.net_g = SynthesizerTrnMs256NSFsid_nono(*cpt["config"])
elif version == "v2": elif self.version == "v2":
if if_f0 == 1: if self.if_f0 == 1:
self.net_g = SynthesizerTrnMs768NSFsid( self.net_g = SynthesizerTrnMs768NSFsid(
*cpt["config"], is_half=config.is_half *cpt["config"], is_half=self.config.is_half
) )
else: else:
self.net_g = SynthesizerTrnMs768NSFsid_nono(*cpt["config"]) self.net_g = SynthesizerTrnMs768NSFsid_nono(*cpt["config"])