Cool_Tools/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of synced 2024-11-27 17:00:54 +01:00
Code Releases Wiki Activity

Format code (#1136)

Browse Source 
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
github-actions[bot] 2023-08-30 19:03:21 +08:00 committed by GitHub
parent 2419d5e1a9
commit ce8177a932
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
configs/config.py
View File

@ -181,7 +181,7 @@ class Config:
) )
except: except:
pass pass
#if self.device != "cpu": # if self.device != "cpu":
import torch_directml import torch_directml
self.device = torch_directml.device(torch_directml.default_device()) self.device = torch_directml.device(torch_directml.default_device())

3
infer-web.py
View File

@ -1,4 +1,5 @@
import os,sys import os, sys
now_dir = os.getcwd() now_dir = os.getcwd()
sys.path.append(now_dir) sys.path.append(now_dir)
import logging import logging

5
infer/lib/rmvpe.py
View File

@ -1,4 +1,4 @@
import pdb,os import pdb, os
import numpy as np import numpy as np
import torch import torch
@ -577,7 +577,8 @@ class RMVPE:
import onnxruntime as ort import onnxruntime as ort
ort_session = ort.InferenceSession( ort_session = ort.InferenceSession(
"%s/rmvpe.onnx"%os.environ["rmvpe_root"], providers=["DmlExecutionProvider"] "%s/rmvpe.onnx" % os.environ["rmvpe_root"],
providers=["DmlExecutionProvider"],
) )
self.model = ort_session self.model = ort_session
else: else:

8
infer/modules/vc/pipeline.py
View File

@ -139,9 +139,13 @@ class Pipeline(object):
if not hasattr(self, "model_rmvpe"): if not hasattr(self, "model_rmvpe"):
from infer.lib.rmvpe import RMVPE from infer.lib.rmvpe import RMVPE
print("loading rmvpe model,%s"%"%s/rmvpe.pt"%os.environ["rmvpe_root"]) print(
"loading rmvpe model,%s" % "%s/rmvpe.pt" % os.environ["rmvpe_root"]
)
self.model_rmvpe = RMVPE( self.model_rmvpe = RMVPE(
"%s/rmvpe.pt"%os.environ["rmvpe_root"], is_half=self.is_half, device=self.device "%s/rmvpe.pt" % os.environ["rmvpe_root"],
is_half=self.is_half,
device=self.device,
) )
f0 = self.model_rmvpe.infer_from_audio(x, thred=0.03) f0 = self.model_rmvpe.infer_from_audio(x, thred=0.03)