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

fixed prompt enhancer import

Browse Source
This commit is contained in:
deepbeepmeep 2025-08-07 12:24:45 +02:00
parent 71d193a13a
commit 3983715eec
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models/wan/wan_handler.py
View File

@ -55,7 +55,7 @@ class family_handler():
"cfg_zero" : True, "cfg_zero" : True,
"cfg_star" : True, "cfg_star" : True,
"adaptive_projected_guidance" : True, "adaptive_projected_guidance" : True,
"skip_steps_cache" : True, "skip_steps_cache" : not (base_model_type in ["i2v_2_2", "ti2v_2_2" ] or "URLs2" in model_def),
}) })
return extra_model_def return extra_model_def

2
wgp.py
View File

@ -4270,7 +4270,7 @@ def generate_video(
if prompt_enhancer_image_caption_model != None and prompt_enhancer !=None and len(prompt_enhancer)>0: if prompt_enhancer_image_caption_model != None and prompt_enhancer !=None and len(prompt_enhancer)>0:
text_encoder_max_tokens = 256 text_encoder_max_tokens = 256
send_cmd("progress", [0, get_latest_status(state, "Enhancing Prompt")]) send_cmd("progress", [0, get_latest_status(state, "Enhancing Prompt")])
from ltx_video.utils.prompt_enhance_utils import generate_cinematic_prompt from models.ltx_video.utils.prompt_enhance_utils import generate_cinematic_prompt
prompt_images = [] prompt_images = []
if "I" in prompt_enhancer: if "I" in prompt_enhancer:
if image_start != None: if image_start != None: