isuneast commited on
Commit
b5369b7
1 Parent(s): 035f2db

rename app.py

Browse files
Files changed (3) hide show
  1. all_process.py +3 -3
  2. webui.py → app.py +0 -0
  3. 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 webui.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,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 webui.py")
665
  else:
666
- kill_specific_process_windows("python webui.py")
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 webui.py`"
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(