This commit is contained in:
Jeremy Jin-Young Kim 2025-06-03 19:55:05 +02:00 committed by GitHub
commit 956d9d5bbf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

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