Skip to content

GitLab

Unverified Commit a1ea3adb authored by Nishant Prabhu's avatar Nishant Prabhu Committed by GitHub
Browse files

Fix LayoutLM ONNX test error (#13710) 

Fix LayoutLM ONNX test error
parent 3a8a8013
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 7 deletions
docs/source/serialization.rst
View file @ a1ea3adb
...@@ -42,6 +42,7 @@ Ready-made configurations include the following models: ...@@ -42,6 +42,7 @@ Ready-made configurations include the following models:
- BERT - BERT
- DistilBERT - DistilBERT
- GPT-2 - GPT-2
- LayoutLM
- RoBERTa - RoBERTa
- T5 - T5
- XLM-RoBERTa - XLM-RoBERTa
......
src/transformers/models/layoutlm/configuration_layoutlm.py
View file @ a1ea3adb
...@@ -183,11 +183,6 @@ class LayoutLMOnnxConfig(OnnxConfig): ...@@ -183,11 +183,6 @@ class LayoutLMOnnxConfig(OnnxConfig):
raise ValueError("Cannot generate dummy inputs without PyTorch installed.") raise ValueError("Cannot generate dummy inputs without PyTorch installed.")
import torch import torch
input_dict["bbox"] = torch.tensor( batch_size, seq_length = input_dict["input_ids"].shape
[ input_dict["bbox"] = torch.tensor([*[box] * seq_length]).tile(batch_size, 1, 1)
[0] * 4,
*[box] * seq_length,
[self.max_2d_positions] * 4,
]
).tile(batch_size, 1, 1)
return input_dict return input_dict
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