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

fixed bug

Browse Source
This commit is contained in:
DeepBeepMeep 2025-06-23 21:54:21 +02:00
parent d001bdedb0
commit dbbb87d075
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
wgp.py
View File

@ -3576,6 +3576,7 @@ def generate_video(
raise gr.Error(f"invalid keep frames {keep_frames_video_guide}") raise gr.Error(f"invalid keep frames {keep_frames_video_guide}")
keep_frames_parsed = keep_frames_parsed[guide_start_frame: guide_start_frame + current_video_length] keep_frames_parsed = keep_frames_parsed[guide_start_frame: guide_start_frame + current_video_length]
context_scale = [ control_net_weight] context_scale = [ control_net_weight]
video_guide_copy2 = video_mask_copy2 = None
if "V" in video_prompt_type: if "V" in video_prompt_type:
process_map = { "Y" : "depth", "W": "scribble", "X": "inpaint", "Z": "flow"} process_map = { "Y" : "depth", "W": "scribble", "X": "inpaint", "Z": "flow"}
process_outside_mask = process_map.get(filter_letters(video_prompt_type, "YWX"), None) process_outside_mask = process_map.get(filter_letters(video_prompt_type, "YWX"), None)
@ -3595,7 +3596,6 @@ def generate_video(
status_info += ", " + process_names[extra_process_list[0]] + " and " + process_names[extra_process_list[1]] status_info += ", " + process_names[extra_process_list[0]] + " and " + process_names[extra_process_list[1]]
send_cmd("progress", [0, get_latest_status(state, status_info)]) send_cmd("progress", [0, get_latest_status(state, status_info)])
video_guide_copy, video_mask_copy = preprocess_video_with_mask(video_guide, video_mask, height=image_size[0], width = image_size[1], max_frames= len(keep_frames_parsed) if guide_start_frame == 0 else len(keep_frames_parsed) - reuse_frames, start_frame = guide_start_frame, fit_canvas = sample_fit_canvas, target_fps = fps, process_type = preprocess_type, expand_scale = mask_expand, RGB_Mask = True, negate_mask = "N" in video_prompt_type, process_outside_mask = process_outside_mask, outpainting_dims = outpainting_dims, proc_no =1 ) video_guide_copy, video_mask_copy = preprocess_video_with_mask(video_guide, video_mask, height=image_size[0], width = image_size[1], max_frames= len(keep_frames_parsed) if guide_start_frame == 0 else len(keep_frames_parsed) - reuse_frames, start_frame = guide_start_frame, fit_canvas = sample_fit_canvas, target_fps = fps, process_type = preprocess_type, expand_scale = mask_expand, RGB_Mask = True, negate_mask = "N" in video_prompt_type, process_outside_mask = process_outside_mask, outpainting_dims = outpainting_dims, proc_no =1 )
video_guide_copy2 = video_mask_copy2 = None
if preprocess_type2 != None: if preprocess_type2 != None:
video_guide_copy2, video_mask_copy2 = preprocess_video_with_mask(video_guide, video_mask, height=image_size[0], width = image_size[1], max_frames= len(keep_frames_parsed) if guide_start_frame == 0 else len(keep_frames_parsed) - reuse_frames, start_frame = guide_start_frame, fit_canvas = sample_fit_canvas, target_fps = fps, process_type = preprocess_type2, expand_scale = mask_expand, RGB_Mask = True, negate_mask = "N" in video_prompt_type, process_outside_mask = process_outside_mask, outpainting_dims = outpainting_dims, proc_no =2 ) video_guide_copy2, video_mask_copy2 = preprocess_video_with_mask(video_guide, video_mask, height=image_size[0], width = image_size[1], max_frames= len(keep_frames_parsed) if guide_start_frame == 0 else len(keep_frames_parsed) - reuse_frames, start_frame = guide_start_frame, fit_canvas = sample_fit_canvas, target_fps = fps, process_type = preprocess_type2, expand_scale = mask_expand, RGB_Mask = True, negate_mask = "N" in video_prompt_type, process_outside_mask = process_outside_mask, outpainting_dims = outpainting_dims, proc_no =2 )
@ -5354,7 +5354,7 @@ def generate_video_tab(update_form = False, state_dict = None, ui_defaults = Non
("Whole Frame", ""), ("Whole Frame", ""),
("Masked Area", "A"), ("Masked Area", "A"),
("Non Masked Area", "NA"), ("Non Masked Area", "NA"),
("Masked Area, rest Outpainted", "XA"), ("Masked Area, rest Inpainted", "XA"),
("Non Masked Area, rest Inpainted", "XNA"), ("Non Masked Area, rest Inpainted", "XNA"),
("Masked Area, rest Depth", "YA"), ("Masked Area, rest Depth", "YA"),
("Non Masked Area, rest Depth", "YNA"), ("Non Masked Area, rest Depth", "YNA"),