fixed End Frames checkbox not always visible

This commit is contained in:
DeepBeepMeep 2025-09-06 00:08:45 +02:00
parent 0649686e9d
commit 66a07db16b
2 changed files with 5 additions and 2 deletions

View File

@ -144,7 +144,9 @@ class family_handler():
if vace_class or base_model_type in ["infinitetalk"]: if vace_class or base_model_type in ["infinitetalk"]:
image_prompt_types_allowed = "TVL" image_prompt_types_allowed = "TVL"
elif base_model_type in ["ti2v_2_2"]: elif base_model_type in ["ti2v_2_2"]:
image_prompt_types_allowed = "TSEV" image_prompt_types_allowed = "TSVL"
elif test_multitalk(base_model_type) or base_model_type in ["fantasy"]:
image_prompt_types_allowed = "SVL"
elif i2v: elif i2v:
image_prompt_types_allowed = "SEVL" image_prompt_types_allowed = "SEVL"
else: else:

3
wgp.py
View File

@ -6623,7 +6623,8 @@ def refresh_image_prompt_type_radio(state, image_prompt_type, image_prompt_type_
image_prompt_type = del_in_sequence(image_prompt_type, "VLTS") image_prompt_type = del_in_sequence(image_prompt_type, "VLTS")
image_prompt_type = add_to_sequence(image_prompt_type, image_prompt_type_radio) image_prompt_type = add_to_sequence(image_prompt_type, image_prompt_type_radio)
any_video_source = len(filter_letters(image_prompt_type, "VL"))>0 any_video_source = len(filter_letters(image_prompt_type, "VL"))>0
end_visible = any_letters(image_prompt_type, "SVL") model_def = get_model_def(state["model_type"])
end_visible = any_letters(image_prompt_type, "SVL") and "E" in model_def.get("image_prompt_types_allowed","")
return image_prompt_type, gr.update(visible = "S" in image_prompt_type ), gr.update(visible = end_visible and ("E" in image_prompt_type) ), gr.update(visible = "V" in image_prompt_type) , gr.update(visible = any_video_source), gr.update(visible = end_visible) return image_prompt_type, gr.update(visible = "S" in image_prompt_type ), gr.update(visible = end_visible and ("E" in image_prompt_type) ), gr.update(visible = "V" in image_prompt_type) , gr.update(visible = any_video_source), gr.update(visible = end_visible)
def refresh_image_prompt_type_endcheckbox(state, image_prompt_type, image_prompt_type_radio, end_checkbox): def refresh_image_prompt_type_endcheckbox(state, image_prompt_type, image_prompt_type_radio, end_checkbox):