initial commit

modeling_tf_my_model.py

@@ -18,3 +18,16 @@ class TFMyModel(TFMyModelPretrainedModel):
         self.n_layers = config.n_layers
         self.hidden_dim = config.hidden_dim
         self.linear = tf.keras.layers.Dense(units=config.n_layers)
+
+    @property
+    def dummy_inputs(self):
+        hidden = tf.zeros(shape=(1, self.config.hidden_dim))
+        dummy_inputs = {"hidden": hidden}
+        return dummy_inputs
+
+    def call(
+        self,
+        hidden,
+    ):
+        self.linear(hidden)
+