initial commit

modeling_tf_my_model.py

@@ -1,4 +1,5 @@
 import tensorflow as tf
+from transformers.modeling_tf_utils import unpack_inputs
 
 from transformers.modeling_tf_utils import TFPreTrainedModel
 
@@ -25,9 +26,11 @@ class TFMyModel(TFMyModelPretrainedModel):
         dummy_inputs = {"hidden": hidden}
         return dummy_inputs
 
+    @unpack_inputs
     def call(
         self,
         hidden,
     ):
+        breakpoint()
         self.linear(hidden)