riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

rename DocumentQuestionAnsweringTool parameter input to match docstring (#23939)

Browse Source 
rename encode input to match docstring
This commit is contained in:
Adam Lewis 2023-06-01 09:54:01 -05:00 committed by GitHub
parent 9193188276
commit 857d4e1c87
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/tools/document_question_answering.py
View File

@ -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}