From 76bceb2fe530a1ae92db85a5416104dbd91de1dd Mon Sep 17 00:00:00 2001
From: root <seahl0119@naver.com>
Date: Fri, 2 May 2025 05:10:43 +0000
Subject: [PATCH] fix: enalbe prompt extend for flf2v

---
 generate.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/generate.py b/generate.py
index 73f273e..d43a5c3 100644
--- 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(