fix: enalbe prompt extend for flf2v

2025-06-05 14:54:54 +00:00 · 2025-05-02 05:10:43 +00:00 · 2025-05-02 05:10:43 +00:00 · 76bceb2fe5
commit 76bceb2fe5
parent 204f899b64
1 changed files with 1 additions and 1 deletions
--- a/generate.py
+++ b/generate.py
@ -272,7 +272,7 @@ def generate(args):
        elif args.prompt_extend_method == "local_qwen":
            prompt_expander = QwenPromptExpander(
                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)
        else:
            raise NotImplementedError(