This commit is contained in:
Jeremy Jin-Young Kim 2025-07-07 11:59:12 -05:00 committed by GitHub
commit 82a2931f11
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(