Add missing key to TFLayoutLM signature (huggingface#28640)

Fix missing bbox in LayoutLM signature
AjayP13 · Jan 22, 2024 · 2fc1cb2 · 2fc1cb2
1 parent ab49da5
commit 2fc1cb2
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/transformers/models/layoutlm/modeling_tf_layoutlm.py b/src/transformers/models/layoutlm/modeling_tf_layoutlm.py
@@ -944,6 +944,12 @@ class TFLayoutLMPreTrainedModel(TFPreTrainedModel):
     config_class = LayoutLMConfig
     base_model_prefix = "layoutlm"
 
+    @property
+    def input_signature(self):
+        signature = super().input_signature
+        signature["bbox"] = tf.TensorSpec(shape=(None, None, 4), dtype=tf.int32, name="bbox")
+        return signature
+
 
 LAYOUTLM_START_DOCSTRING = r"""