This commit is contained in:
Jeremy Jin-Young Kim 2025-06-06 16:00:12 +09:00 committed by GitHub
commit a7d68f0c31
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": 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(