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

fix incorrect internal resize bug

Browse Source
This commit is contained in:
deepbeepmeep 2025-08-25 13:28:09 +02:00
parent 485aa93102
commit 4dacbcb4ad
3 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models/hyvideo/hunyuan_handler.py
View File

@ -173,5 +173,5 @@ class family_handler():
"remove_background_images_ref": 0, "remove_background_images_ref": 0,
"skip_steps_start_step_perc": 25, "skip_steps_start_step_perc": 25,
"video_length": 129, "video_length": 129,
"video_prompt_type": "I", "video_prompt_type": "KI",
}) })

1
models/wan/wan_handler.py
View File

@ -101,7 +101,6 @@ class family_handler():
"adaptive_projected_guidance" : True, "adaptive_projected_guidance" : True,
"tea_cache" : not (base_model_type in ["i2v_2_2", "ti2v_2_2" ] or multiple_submodels), "tea_cache" : not (base_model_type in ["i2v_2_2", "ti2v_2_2" ] or multiple_submodels),
"mag_cache" : True, "mag_cache" : True,
"first_ref_is_start_image": base_model_type in ["infinitetalk"],
"sample_solvers":[ "sample_solvers":[
("unipc", "unipc"), ("unipc", "unipc"),
("euler", "euler"), ("euler", "euler"),

16
wgp.py
View File

@ -55,8 +55,8 @@ AUTOSAVE_FILENAME = "queue.zip"
PROMPT_VARS_MAX = 10 PROMPT_VARS_MAX = 10
target_mmgp_version = "3.5.10" target_mmgp_version = "3.5.10"
WanGP_version = "8.1" WanGP_version = "8.11"
settings_version = 2.25 settings_version = 2.26
max_source_video_frames = 3000 max_source_video_frames = 3000
prompt_enhancer_image_caption_model, prompt_enhancer_image_caption_processor, prompt_enhancer_llm_model, prompt_enhancer_llm_tokenizer = None, None, None, None prompt_enhancer_image_caption_model, prompt_enhancer_image_caption_processor, prompt_enhancer_llm_model, prompt_enhancer_llm_tokenizer = None, None, None, None
@ -2073,7 +2073,10 @@ def fix_settings(model_type, ui_defaults):
video_prompt_type = video_prompt_type.replace("I", "KI") video_prompt_type = video_prompt_type.replace("I", "KI")
if remove_background_images_ref != 0: if remove_background_images_ref != 0:
remove_background_images_ref = 1 remove_background_images_ref = 1
if base_model_type in ["hunyuan_avatar"]: remove_background_images_ref = 0 if base_model_type in ["hunyuan_avatar"]:
remove_background_images_ref = 0
if settings_version < 2.26:
if not "K" in video_prompt_type: video_prompt_type = video_prompt_type.replace("I", "KI")
ui_defaults["remove_background_images_ref"] = remove_background_images_ref ui_defaults["remove_background_images_ref"] = remove_background_images_ref
ui_defaults["video_prompt_type"] = video_prompt_type ui_defaults["video_prompt_type"] = video_prompt_type
@ -4488,6 +4491,9 @@ def generate_video(
frames_to_inject = [] frames_to_inject = []
any_background_ref = False any_background_ref = False
outpainting_dims = None if video_guide_outpainting== None or len(video_guide_outpainting) == 0 or video_guide_outpainting == "0 0 0 0" or video_guide_outpainting.startswith("#") else [int(v) for v in video_guide_outpainting.split(" ")] outpainting_dims = None if video_guide_outpainting== None or len(video_guide_outpainting) == 0 or video_guide_outpainting == "0 0 0 0" or video_guide_outpainting.startswith("#") else [int(v) for v in video_guide_outpainting.split(" ")]
# Output Video Ratio Priorities:
# Source Video or Start Image > Control Video > Image Ref (background or positioned frames only) > UI Width, Height
# Image Ref (non background and non positioned frames) are boxed in a white canvas in order to keep their own width/height ratio
if image_refs is not None and len(image_refs) > 0: if image_refs is not None and len(image_refs) > 0:
frames_positions_list = [ int(pos)-1 for pos in frames_positions.split(" ")] if frames_positions is not None and len(frames_positions)> 0 else [] frames_positions_list = [ int(pos)-1 for pos in frames_positions.split(" ")] if frames_positions is not None and len(frames_positions)> 0 else []
@ -4510,7 +4516,7 @@ def generate_video(
send_cmd("progress", [0, get_latest_status(state, "Removing Images References Background")]) send_cmd("progress", [0, get_latest_status(state, "Removing Images References Background")])
os.environ["U2NET_HOME"] = os.path.join(os.getcwd(), "ckpts", "rembg") os.environ["U2NET_HOME"] = os.path.join(os.getcwd(), "ckpts", "rembg")
from shared.utils.utils import resize_and_remove_background from shared.utils.utils import resize_and_remove_background
image_refs[nb_frames_positions:] = resize_and_remove_background(image_refs[nb_frames_positions:] , width, height, remove_background_images_ref > 0, any_background_ref, fit_into_canvas= False if (vace or hunyuan_avatar or flux or qwen) else fit_canvas ) # no fit for vace ref images as it is done later image_refs[nb_frames_positions:] = resize_and_remove_background(image_refs[nb_frames_positions:] , width, height, remove_background_images_ref > 0, any_background_ref, fit_into_canvas= not (any_background_ref or vace) ) # no fit for vace ref images as it is done later
update_task_thumbnails(task, locals()) update_task_thumbnails(task, locals())
send_cmd("output") send_cmd("output")
joint_pass = boost ==1 #and profile != 1 and profile != 3 joint_pass = boost ==1 #and profile != 1 and profile != 3
@ -7192,7 +7198,7 @@ def generate_video_tab(update_form = False, state_dict = None, ui_defaults = Non
else: else:
video_prompt_type_image_refs = gr.Dropdown( video_prompt_type_image_refs = gr.Dropdown(
choices=[ ("Start", "KI"),("Ref Image", "I")], choices=[ ("Start", "KI"),("Ref Image", "I")],
value="KI" if model_def.get("first_ref_is_start_image", False) else "I", value=filter_letters(video_prompt_type_value, "KI"),
visible = False, visible = False,
label="Start / Reference Images", scale = 2 label="Start / Reference Images", scale = 2
) )