Function build_wan21_components

app/utils/model_components.py:62–168

构建 wan2.1 模型相关组件 必须在 Gradio 的 with 块内调用 Returns: dict: 包含所有 wan2.1 相关组件的字典

(model_path, model_path_input, model_type_input, task_type_input, download_source_input, update_funcs, download_funcs, lang="zh")

def build_wan21_components(model_path, model_path_input, model_type_input, task_type_input, download_source_input, update_funcs, download_funcs, lang="zh"):
    """构建 wan2.1 模型相关组件
    必须在 Gradio 的 with 块内调用

    Returns:
        dict: 包含所有 wan2.1 相关组件的字典
    """
    # wan2.1：Diffusion模型
    with gr.Column(elem_classes=["diffusion-model-group"]) as wan21_row:
        with gr.Row():
            with gr.Column(scale=5):
                dit_choices_init = get_dit_choices(model_path, "wan2.1", "i2v")
                dit_path_input = gr.Dropdown(
                    label="🎨 Diffusion模型",
                    choices=dit_choices_init,
                    value=dit_choices_init[0] if dit_choices_init else "",
                    allow_custom_value=True,
                    visible=True,
                )
            with gr.Column(scale=1, min_width=150):
                # 初始化时检查模型状态，确定下载按钮的初始可见性
                from utils.model_utils import check_model_exists, extract_model_name

                dit_btn_visible = False
                if dit_choices_init:
                    first_choice = dit_choices_init[0]
                    actual_name = extract_model_name(first_choice)
                    dit_exists = check_model_exists(model_path, actual_name)
                    dit_btn_visible = not dit_exists

                dit_download_btn = gr.Button("📥 下载", visible=dit_btn_visible, size="sm", variant="secondary")
        dit_download_status = gr.Markdown("", visible=False)

        lora_choices_init = get_lora_choices(model_path)
        with gr.Row():
            with gr.Column(scale=1):
                use_lora = gr.Checkbox(
                    label=t("use_lora", lang),
                    value=False,
                )
                lora_path_input = gr.Dropdown(
                    label=t("lora", lang),
                    choices=lora_choices_init,
                    value=lora_choices_init[0] if lora_choices_init and lora_choices_init[0] else "",
                    allow_custom_value=True,
                    visible=False,
                    info=t("lora_info", lang),
                )
                lora_strength = gr.Slider(
                    label=t("lora_strength", lang),
                    minimum=0.0,
                    maximum=10.0,
                    step=0.1,
                    value=1.0,
                    visible=False,
                    info=t("lora_strength_info", lang),
                )