Remove @tf.keras.utils.register_keras_serializable(package="Text") as the...

Remove @tf.keras.utils.register_keras_serializable(package="Text") as the model cannot be serialized anyway. PiperOrigin-RevId: 408904971

Remove @tf.keras.utils.register_keras_serializable(package="Text") as the...
Remove @tf.keras.utils.register_keras_serializable(package="Text") as the model cannot be serialized anyway. PiperOrigin-RevId: 408904971
a241b9ae · Hongkun Yu · A. Unique TensorFlower · 3c72f65f · a241b9ae
Commit a241b9ae authored Nov 10, 2021 by Hongkun Yu Committed by A. Unique TensorFlower Nov 10, 2021
Show whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

official/nlp/modeling/models/seq2seq_transformer.py official/nlp/modeling/models/seq2seq_transformer.py +0 -1

No files found.
--- a/official/nlp/modeling/models/seq2seq_transformer.py
+++ b/official/nlp/modeling/models/seq2seq_transformer.py
@@ -26,7 +26,6 @@ from official.nlp.modeling.ops import beam_search
 EOS_ID = 1


-@tf.keras.utils.register_keras_serializable(package="Text")
 class Seq2SeqTransformer(tf.keras.Model):
  """Transformer model with Keras.