This commit is contained in:
Jeremy Jin-Young Kim 2025-05-15 13:01:27 +08:00 committed by GitHub
commit d3a0e7b077
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

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