Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-11-04 14:16:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix control image test

Browse Source
This commit is contained in:
deepbeepmeep 2025-08-04 13:08:32 +02:00
parent c738462831
commit a989d9d2c0
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
wgp.py
View File

@ -177,7 +177,7 @@ def process_prompt_and_add_tasks(state, model_choice):
queue = gen.get("queue", []) queue = gen.get("queue", [])
return get_queue_table(queue) return get_queue_table(queue)
model_def = get_model_def(model_type) model_def = get_model_def(model_type)
image_outputs = model_def.get("image_outputs", False) image_outputs = inputs["image_mode"] == 1
no_steps_skipping = model_def.get("no_steps_skipping", False) no_steps_skipping = model_def.get("no_steps_skipping", False)
model_type = get_base_model_type(model_type) model_type = get_base_model_type(model_type)
inputs["model_filename"] = model_filename inputs["model_filename"] = model_filename
@ -7087,20 +7087,21 @@ def generate_video_tab(update_form = False, state_dict = None, ui_defaults = Non
label="Video to Video", scale = 2, show_label= False, visible= True label="Video to Video", scale = 2, show_label= False, visible= True
) )
elif vace : elif vace :
pose_label = "Pose" if image_outputs else "Motion"
video_prompt_type_video_guide = gr.Dropdown( video_prompt_type_video_guide = gr.Dropdown(
choices=[ choices=[
("No Control Image" if image_outputs else "No Control Video", ""), ("No Control Image" if image_outputs else "No Control Video", ""),
("Keep Control Image Unchanged" if image_outputs else "Keep Control Video Unchanged", "UV"), ("Keep Control Image Unchanged" if image_outputs else "Keep Control Video Unchanged", "UV"),
("Transfer Human Motion", "PV"), (f"Transfer Human {pose_label}" , "PV"),
("Transfer Depth", "DV"), ("Transfer Depth", "DV"),
("Transfer Shapes", "SV"), ("Transfer Shapes", "SV"),
("Transfer Flow", "LV"), ("Transfer Flow", "LV"),
("Recolorize", "CV"), ("Recolorize", "CV"),
("Perform Inpainting", "MV"), ("Perform Inpainting", "MV"),
("Use Vace raw format", "V"), ("Use Vace raw format", "V"),
("Transfer Human Motion & Depth", "PDV"), (f"Transfer Human {pose_label} & Depth", "PDV"),
("Transfer Human Motion & Shapes", "PSV"), (f"Transfer Human {pose_label} & Shapes", "PSV"),
("Transfer Human Motion & Flow", "PLV"), (f"Transfer Human {pose_label} & Flow", "PLV"),
("Transfer Depth & Shapes", "DSV"), ("Transfer Depth & Shapes", "DSV"),
("Transfer Depth & Flow", "DLV"), ("Transfer Depth & Flow", "DLV"),
("Transfer Shapes & Flow", "SLV"), ("Transfer Shapes & Flow", "SLV"),