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

fix: enalbe prompt extend for flf2v

Browse Source
This commit is contained in:
root 2025-05-02 05:10:43 +00:00
parent 204f899b64
commit 76bceb2fe5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
generate.py
View File

@ -272,7 +272,7 @@ def generate(args):
elif args.prompt_extend_method == "local_qwen": elif args.prompt_extend_method == "local_qwen":
prompt_expander = QwenPromptExpander( prompt_expander = QwenPromptExpander(
model_name=args.prompt_extend_model, model_name=args.prompt_extend_model,
is_vl="i2v" in args.task, is_vl="i2v" in args.task or "flf2v" in args.task,
device=rank) device=rank)
else: else:
raise NotImplementedError( raise NotImplementedError(