Commit 566acef5 authored by lintangsutawika's avatar lintangsutawika
Browse files

update multimodal args

parent df137b46
...@@ -1316,11 +1316,18 @@ class ConfigurableTask(Task): ...@@ -1316,11 +1316,18 @@ class ConfigurableTask(Task):
elif self.OUTPUT_TYPE == "generate_until": elif self.OUTPUT_TYPE == "generate_until":
arguments = (ctx, deepcopy(self.config.generation_kwargs)) arguments = (ctx, deepcopy(self.config.generation_kwargs))
multimodal_arg = {}
if self.doc_to_visual: if self.doc_to_visual:
multimodal_arg = {
**multimodal_arg,
**{"visual": self.doc_to_visual(doc)},
}
if bool(multimodal_arg):
if isinstance(arguments, list): if isinstance(arguments, list):
arguments = [arg + (self.doc_to_visual(doc),) for arg in arguments] arguments = [arg + (multimodal_arg,) for arg in arguments]
else: else:
arguments = arguments + (self.doc_to_visual(doc),) arguments = arguments + (multimodal_arg,)
if isinstance(arguments, type): if isinstance(arguments, type):
if aux_arguments is not None: if aux_arguments is not None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment