jshiffer/Retrieval-based-Voice-Conversion
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add type declarations

Browse Source
This commit is contained in:
Ftps
2023-11-17 18:24:37 +09:00
parent 7a6ce09b6f
commit 320b2bfef8
1 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
rvc/configs/config.py
View File

@@ -17,7 +17,7 @@ except (ImportError, Exception):
pass pass
import logging import logging
logger = logging.getLogger(__name__) logger: logging.Logger = logging.getLogger(__name__)
version_config_list: list = [ version_config_list: list = [
@@ -35,13 +35,14 @@ class Config:
return cls._instance return cls._instance
def __init__(self): def __init__(self):
self.device = "cuda:0" self.device: str = "cuda:0"
self.is_half = True self.is_half: bool = True
self.use_jit = False self.use_jit: bool = False
self.n_cpu = 0 self.n_cpu: int = cpu_count()
self.gpu_name = None self.gpu_name: str | None = None
self.json_config = self.load_config_json() self.json_config = self.load_config_json()
self.gpu_mem = None self.gpu_mem: int | None = None
self.instead: str | None = None
( (
self.python_cmd, self.python_cmd,
self.listen_port, self.listen_port,
@@ -60,8 +61,7 @@ class Config:
@staticmethod @staticmethod
def arg_parse() -> tuple: def arg_parse() -> tuple:
exe = sys.executable or "python" parser: argparse.ArgumentParser = argparse.ArgumentParser()
parser = argparse.ArgumentParser()
parser.add_argument("--port", type=int, default=7865, help="Listen port") parser.add_argument("--port", type=int, default=7865, help="Listen port")
parser.add_argument( parser.add_argument(
"--pycmd", "--pycmd",
@@ -82,7 +82,7 @@ class Config:
action="store_true", action="store_true",
help="torch_dml", help="torch_dml",
) )
cmd_opts = parser.parse_args() cmd_opts: argparse.Namespace = parser.parse_args()
cmd_opts.port = cmd_opts.port if 0 <= cmd_opts.port <= 65535 else 7865 cmd_opts.port = cmd_opts.port if 0 <= cmd_opts.port <= 65535 else 7865