Unverified Commit bf674153 authored by Matt's avatar Matt Committed by GitHub
Browse files

Add missing key to TFLayoutLM signature (#28640)

Fix missing bbox in LayoutLM signature
parent f0acf7b6
...@@ -944,6 +944,12 @@ class TFLayoutLMPreTrainedModel(TFPreTrainedModel): ...@@ -944,6 +944,12 @@ class TFLayoutLMPreTrainedModel(TFPreTrainedModel):
config_class = LayoutLMConfig config_class = LayoutLMConfig
base_model_prefix = "layoutlm" 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""" LAYOUTLM_START_DOCSTRING = r"""
......
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