PublicMirrors/voice-changer
1
0
Fork 0
mirror of https://github.com/w-okada/voice-changer.git synced 2025-01-23 21:45:00 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add ModelType for valid model type names

Browse Source
This commit is contained in:
frodo821 2023-04-10 09:05:14 +09:00
parent 138d17719c
commit adcc0a2d86
No known key found for this signature in database
GPG Key ID: 29F87B8DC3A2425D
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
server/const.py
View File

@ -1,6 +1,10 @@
import os import os
import sys import sys
import tempfile import tempfile
from typing import Literal, TypeAlias
ModelType: TypeAlias = Literal['MMVCv15', 'MMVCv13', 'so-vits-svc-40v2', 'so-vits-svc-40', 'so-vits-svc-40_c', 'DDSP-SVC', 'RVC']
ERROR_NO_ONNX_SESSION = "ERROR_NO_ONNX_SESSION" ERROR_NO_ONNX_SESSION = "ERROR_NO_ONNX_SESSION"
@ -22,14 +26,14 @@ TMP_DIR = os.path.join(tmpdir.name, "tmp_dir") if hasattr(sys, "_MEIPASS") else
os.makedirs(TMP_DIR, exist_ok=True) os.makedirs(TMP_DIR, exist_ok=True)
modelType = "MMVCv15" modelType: ModelType = "MMVCv15"
def getModelType(): def getModelType() -> ModelType:
return modelType return modelType
def setModelType(_modelType: str): def setModelType(_modelType: ModelType):
global modelType global modelType
modelType = _modelType modelType = _modelType