"tools/git@developer.sourcefind.cn:OpenDAS/torchani.git" did not exist on "5bac472d635a5182c62b55cef3a29d1c9788c3f0"
Unverified Commit 857d4e1c authored by Adam Lewis's avatar Adam Lewis Committed by GitHub
Browse files

rename DocumentQuestionAnsweringTool parameter input to match docstring (#23939)

rename encode input to match docstring
parent 91931882
...@@ -46,13 +46,13 @@ class DocumentQuestionAnsweringTool(PipelineTool): ...@@ -46,13 +46,13 @@ class DocumentQuestionAnsweringTool(PipelineTool):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
def encode(self, image: "Image", question: str): def encode(self, document: "Image", question: str):
task_prompt = "<s_docvqa><s_question>{user_input}</s_question><s_answer>" task_prompt = "<s_docvqa><s_question>{user_input}</s_question><s_answer>"
prompt = task_prompt.replace("{user_input}", question) prompt = task_prompt.replace("{user_input}", question)
decoder_input_ids = self.pre_processor.tokenizer( decoder_input_ids = self.pre_processor.tokenizer(
prompt, add_special_tokens=False, return_tensors="pt" prompt, add_special_tokens=False, return_tensors="pt"
).input_ids ).input_ids
pixel_values = self.pre_processor(image, return_tensors="pt").pixel_values pixel_values = self.pre_processor(document, return_tensors="pt").pixel_values
return {"decoder_input_ids": decoder_input_ids, "pixel_values": pixel_values} return {"decoder_input_ids": decoder_input_ids, "pixel_values": pixel_values}
......
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