Compare commits

...

2 Commits

Author SHA1 Message Date
Jeremy Jin-Young Kim
956d9d5bbf
Merge 76bceb2fe5 into ec902046f6 2025-06-03 19:55:05 +02:00
root
76bceb2fe5 fix: enalbe prompt extend for flf2v 2025-05-02 05:10:43 +00:00

View File

@ -312,7 +312,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(