-
Notifications
You must be signed in to change notification settings - Fork 117
/
app.py
69 lines (61 loc) · 3.36 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
from scripts.modeldownloader import install_deepspeed_based_on_python_version
from argparse import ArgumentParser
from loguru import logger
import os
import subprocess
parser = ArgumentParser(description="Run the Uvicorn server.")
parser.add_argument("-hs", "--host", default="127.0.0.1", help="Host to bind")
parser.add_argument("-p", "--port", default=8010,
type=int, help="Port to bind")
parser.add_argument("-d", "--device", default="cuda", type=str,
help="Device that will be used, you can choose cpu or cuda")
parser.add_argument("-sf", "--speaker_folder", default="speakers/",
type=str, help="The folder where you get the samples for tts")
parser.add_argument("-o", "--output", default="output/",
type=str, help="Output folder")
parser.add_argument("-l", "--language", default="Auto",
type=str, help="WebUI language")
parser.add_argument("-ms", "--model-source", default="local", choices=["api", "local"],
help="Define the model source: 'api' for latest version from repository, api inference or 'local' for using local inference and model v2.0.2.")
parser.add_argument("-v", "--version", default="v2.0.2", type=str,
help="You can specify which version of xtts to use,This version will be used everywhere in local, api and apiManual.")
parser.add_argument("--lowvram", action='store_true',
help="Enable low vram mode which switches the model to RAM when not actively processing.")
parser.add_argument("--deepspeed", action='store_true',
help="Enable deepspeed acceleration, works on windows on python 3.10 and 3.11.")
parser.add_argument("--share", action='store_true',
help="Allows the interface to be used outside the local computer.")
parser.add_argument("--rvc", action='store_true',
help="Choose extensions what you add to main webui [-e rvc]")
args = parser.parse_args()
# Set environment variable for output folder.
os.environ['DEVICE'] = args.device
# Set environment variable for output folder.
os.environ['OUTPUT'] = args.output
# Set environment variable for speaker folder.
os.environ['SPEAKER'] = args.speaker_folder
os.environ['BASE_URL'] = "http://" + args.host + ":" + \
str(args.port) # Set environment variable for base url."
# Set environment variable for the model source
os.environ['MODEL_SOURCE'] = args.model_source
os.environ["LOWVRAM_MODE"] = str(args.lowvram).lower() # Set lowvram mode
os.environ["DEEPSPEED"] = str(args.deepspeed).lower() # Enable Streaming mode
os.environ["MODEL_VERSION"] = args.version # Specify version of XTTS model
os.environ["LANGUAGE"] = args.language
os.environ["RVC_ENABLED"] = str(args.rvc).lower()
# Install RVC
if args.rvc:
# Create rvc folder
os.makedirs("rvc", exist_ok=True)
if (not os.path.exists("venv/rvc_venv")):
logger.info("Installing RVC and OpenVoice libraries...")
subprocess.run(["python", "scripts/install_rvc_venv.py"])
logger.info("RVC and OpenVoice installation is complete")
# else:
# logger.info("RVC libraries are already installed")
# Check deepspeed
install_deepspeed_based_on_python_version()
if __name__ == "__main__":
from xtts_webui import demo
demo.launch(share=args.share, inbrowser=True,
server_name=args.host, server_port=args.port)