From a1ea3adb28c5eb6c0b58d36d97a46360cf143213 Mon Sep 17 00:00:00 2001
From: Nishant Prabhu <33579638+nishprabhu@users.noreply.github.com>
Date: Wed, 29 Sep 2021 19:20:15 +0530
Subject: [PATCH] Fix LayoutLM ONNX test error (#13710)

Fix LayoutLM ONNX test error
---
 docs/source/serialization.rst                            | 1 +
 .../models/layoutlm/configuration_layoutlm.py            | 9 ++-------
 2 files changed, 3 insertions(+), 7 deletions(-)

diff --git a/docs/source/serialization.rst b/docs/source/serialization.rst
index fada8bd98e..285e24cb56 100644
--- a/docs/source/serialization.rst
+++ b/docs/source/serialization.rst
@@ -42,6 +42,7 @@ Ready-made configurations include the following models:
 - BERT
 - DistilBERT
 - GPT-2
+- LayoutLM
 - RoBERTa
 - T5
 - XLM-RoBERTa
diff --git a/src/transformers/models/layoutlm/configuration_layoutlm.py b/src/transformers/models/layoutlm/configuration_layoutlm.py
index 61a6ce264d..913a6bf792 100644
--- a/src/transformers/models/layoutlm/configuration_layoutlm.py
+++ b/src/transformers/models/layoutlm/configuration_layoutlm.py
@@ -183,11 +183,6 @@ class LayoutLMOnnxConfig(OnnxConfig):
             raise ValueError("Cannot generate dummy inputs without PyTorch installed.")
         import torch
 
-        input_dict["bbox"] = torch.tensor(
-            [
-                [0] * 4,
-                *[box] * seq_length,
-                [self.max_2d_positions] * 4,
-            ]
-        ).tile(batch_size, 1, 1)
+        batch_size, seq_length = input_dict["input_ids"].shape
+        input_dict["bbox"] = torch.tensor([*[box] * seq_length]).tile(batch_size, 1, 1)
         return input_dict