initial commit

modeling_tf_my_model.py

@@ -2,7 +2,7 @@ import tensorflow as tf
 
 from transformers.modeling_tf_utils import TFPreTrainedModel
 
-from configuration_my_model import MyModelConfig
+from .configuration_my_model import MyModelConfig
 
 
 class TFMyModelPretrainedModel(TFPreTrainedModel):
@@ -18,9 +18,3 @@ 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)
-
-
-config = MyModelConfig()
-model = TFMyModel(config)
-print(model)
-model.save_pretrained("my_model")