| | import gradio as gr |
| | from app.original import * |
| |
|
| |
|
| | def ckpt_edits(): |
| | with gr.TabItem(i18n("ckpt处理")): |
| | with gr.Group(): |
| | gr.Markdown(value=i18n("模型融合, 可用于测试音色融合")) |
| | with gr.Row(): |
| | ckpt_a = gr.Textbox( |
| | label=i18n("A模型路径"), value="", interactive=True |
| | ) |
| | ckpt_b = gr.Textbox( |
| | label=i18n("B模型路径"), value="", interactive=True |
| | ) |
| | alpha_a = gr.Slider( |
| | minimum=0, |
| | maximum=1, |
| | label=i18n("A模型权重"), |
| | value=0.5, |
| | interactive=True, |
| | ) |
| | with gr.Row(): |
| | sr_ = gr.Radio( |
| | label=i18n("目标采样率"), |
| | choices=["40k", "48k"], |
| | value="40k", |
| | interactive=True, |
| | ) |
| | if_f0_ = gr.Radio( |
| | label=i18n("模型是否带音高指导"), |
| | choices=[i18n("是"), i18n("否")], |
| | value=i18n("是"), |
| | interactive=True, |
| | ) |
| | info__ = gr.Textbox( |
| | label=i18n("要置入的模型信息"), |
| | value="", |
| | max_lines=8, |
| | interactive=True, |
| | ) |
| | name_to_save0 = gr.Textbox( |
| | label=i18n("保存的模型名不带后缀"), |
| | value="", |
| | max_lines=1, |
| | interactive=True, |
| | ) |
| | version_2 = gr.Radio( |
| | label=i18n("模型版本型号"), |
| | choices=["v1", "v2"], |
| | value="v1", |
| | interactive=True, |
| | ) |
| | with gr.Row(): |
| | but6 = gr.Button(i18n("融合"), variant="primary") |
| | info4 = gr.Textbox(label=i18n("输出信息"), value="", max_lines=8) |
| | but6.click( |
| | merge, |
| | [ |
| | ckpt_a, |
| | ckpt_b, |
| | alpha_a, |
| | sr_, |
| | if_f0_, |
| | info__, |
| | name_to_save0, |
| | version_2, |
| | ], |
| | info4, |
| | api_name="ckpt_merge", |
| | ) |
| | with gr.Group(): |
| | gr.Markdown( |
| | value=i18n("修改模型信息(仅支持weights文件夹下提取的小模型文件)") |
| | ) |
| | with gr.Row(): |
| | ckpt_path0 = gr.Textbox( |
| | label=i18n("模型路径"), value="", interactive=True |
| | ) |
| | info_ = gr.Textbox( |
| | label=i18n("要改的模型信息"), |
| | value="", |
| | max_lines=8, |
| | interactive=True, |
| | ) |
| | name_to_save1 = gr.Textbox( |
| | label=i18n("保存的文件名, 默认空为和源文件同名"), |
| | value="", |
| | max_lines=8, |
| | interactive=True, |
| | ) |
| | with gr.Row(): |
| | but7 = gr.Button(i18n("修改"), variant="primary") |
| | info5 = gr.Textbox(label=i18n("输出信息"), value="", max_lines=8) |
| | but7.click( |
| | change_info, |
| | [ckpt_path0, info_, name_to_save1], |
| | info5, |
| | api_name="ckpt_modify", |
| | ) |
| | with gr.Group(): |
| | gr.Markdown( |
| | value=i18n("查看模型信息(仅支持weights文件夹下提取的小模型文件)") |
| | ) |
| | with gr.Row(): |
| | ckpt_path1 = gr.Textbox( |
| | label=i18n("模型路径"), value="", interactive=True |
| | ) |
| | but8 = gr.Button(i18n("查看"), variant="primary") |
| | info6 = gr.Textbox(label=i18n("输出信息"), value="", max_lines=8) |
| | but8.click(show_info, [ckpt_path1], info6, api_name="ckpt_show") |
| | with gr.Group(): |
| | gr.Markdown( |
| | value=i18n( |
| | "模型提取(输入logs文件夹下大文件模型路径),适用于训一半不想训了模型没有自动提取保存小文件模型,或者想测试中间模型的情况" |
| | ) |
| | ) |
| | with gr.Row(): |
| | ckpt_path2 = gr.Textbox( |
| | label=i18n("模型路径"), |
| | value="E:\\codes\\py39\\logs\\mi-test_f0_48k\\G_23333.pth", |
| | interactive=True, |
| | ) |
| | save_name = gr.Textbox( |
| | label=i18n("保存名"), value="", interactive=True |
| | ) |
| | sr__ = gr.Radio( |
| | label=i18n("目标采样率"), |
| | choices=["32k", "40k", "48k"], |
| | value="40k", |
| | interactive=True, |
| | ) |
| | if_f0__ = gr.Radio( |
| | label=i18n("模型是否带音高指导,1是0否"), |
| | choices=["1", "0"], |
| | value="1", |
| | interactive=True, |
| | ) |
| | version_1 = gr.Radio( |
| | label=i18n("模型版本型号"), |
| | choices=["v1", "v2"], |
| | value="v2", |
| | interactive=True, |
| | ) |
| | info___ = gr.Textbox( |
| | label=i18n("要置入的模型信息"), |
| | value="", |
| | max_lines=8, |
| | interactive=True, |
| | ) |
| | but9 = gr.Button(i18n("提取"), variant="primary") |
| | info7 = gr.Textbox(label=i18n("输出信息"), value="", max_lines=8) |
| | ckpt_path2.change( |
| | change_info_, [ckpt_path2], [sr__, if_f0__, version_1] |
| | ) |
| | but9.click( |
| | extract_small_model, |
| | [ckpt_path2, save_name, sr__, if_f0__, info___, version_1], |
| | info7, |
| | api_name="ckpt_extract", |
| | ) |
