rename app.py
Browse files- all_process.py +3 -3
- webui.py → app.py +0 -0
- webui_preprocess.py +1 -1
all_process.py
CHANGED
@@ -579,7 +579,7 @@ def do_webui_infer():
|
|
579 |
comp_c_path = os.path.join(os.path.abspath(data_path), config_path)
|
580 |
if os.path.exists(comp_c_path) and os.path.exists(comp_m_path):
|
581 |
webui_port = yml["webui"]["port"]
|
582 |
-
subprocess.Popen("python
|
583 |
url = f"http://localhost:{webui_port} | http://127.0.0.1:{webui_port}"
|
584 |
msg = f"推理端已开启, 到控制台中复制网址打开页面\n{url}\n选择的模型:{model_path}"
|
585 |
logger.info(msg)
|
@@ -661,9 +661,9 @@ def stop_webui_infer():
|
|
661 |
yml = load_yaml_data_in_fact()
|
662 |
webui_port = yml["webui"]["port"]
|
663 |
if platform.system() == "Linux":
|
664 |
-
kill_specific_process_linux("python
|
665 |
else:
|
666 |
-
kill_specific_process_windows("python
|
667 |
msg = f"尝试终止推理进程,请到控制台查看情况\nport={webui_port}"
|
668 |
logger.critical(msg)
|
669 |
return gr.Textbox(value=msg)
|
|
|
579 |
comp_c_path = os.path.join(os.path.abspath(data_path), config_path)
|
580 |
if os.path.exists(comp_c_path) and os.path.exists(comp_m_path):
|
581 |
webui_port = yml["webui"]["port"]
|
582 |
+
subprocess.Popen("python app.py", shell=True)
|
583 |
url = f"http://localhost:{webui_port} | http://127.0.0.1:{webui_port}"
|
584 |
msg = f"推理端已开启, 到控制台中复制网址打开页面\n{url}\n选择的模型:{model_path}"
|
585 |
logger.info(msg)
|
|
|
661 |
yml = load_yaml_data_in_fact()
|
662 |
webui_port = yml["webui"]["port"]
|
663 |
if platform.system() == "Linux":
|
664 |
+
kill_specific_process_linux("python app.py")
|
665 |
else:
|
666 |
+
kill_specific_process_windows("python app.py")
|
667 |
msg = f"尝试终止推理进程,请到控制台查看情况\nport={webui_port}"
|
668 |
logger.critical(msg)
|
669 |
return gr.Textbox(value=msg)
|
webui.py → app.py
RENAMED
File without changes
|
webui_preprocess.py
CHANGED
@@ -162,7 +162,7 @@ if __name__ == "__main__":
|
|
162 |
value="## 训练模型及部署:\n"
|
163 |
"修改根目录下的 `config.yml` 中 `dataset_path` 一项为 `data/{你的数据集名称}`\n"
|
164 |
"- 训练:将[预训练模型文件](https://openi.pcl.ac.cn/Stardust_minus/Bert-VITS2/modelmanage/show_model)(`D_0.pth`、`DUR_0.pth` 和 `G_0.pth`)放到 `data/{你的数据集名称}/models` 文件夹下,执行 `torchrun --nproc_per_node=1 train_ms.py` 命令(多卡运行可参考 `run_MnodesAndMgpus.sh` 中的命令。\n"
|
165 |
-
"- 部署:修改根目录下的 `config.yml` 中 `webui` 下 `model` 一项为 `models/{权重文件名}.pth` (如 G_10000.pth),然后执行 `python
|
166 |
)
|
167 |
|
168 |
generate_config_btn.click(
|
|
|
162 |
value="## 训练模型及部署:\n"
|
163 |
"修改根目录下的 `config.yml` 中 `dataset_path` 一项为 `data/{你的数据集名称}`\n"
|
164 |
"- 训练:将[预训练模型文件](https://openi.pcl.ac.cn/Stardust_minus/Bert-VITS2/modelmanage/show_model)(`D_0.pth`、`DUR_0.pth` 和 `G_0.pth`)放到 `data/{你的数据集名称}/models` 文件夹下,执行 `torchrun --nproc_per_node=1 train_ms.py` 命令(多卡运行可参考 `run_MnodesAndMgpus.sh` 中的命令。\n"
|
165 |
+
"- 部署:修改根目录下的 `config.yml` 中 `webui` 下 `model` 一项为 `models/{权重文件名}.pth` (如 G_10000.pth),然后执行 `python app.py`"
|
166 |
)
|
167 |
|
168 |
generate_config_btn.click(
|