Internal change

PiperOrigin-RevId: 401894268

Internal change
PiperOrigin-RevId: 401894268
dc7a68eb · Frederick Liu · A. Unique TensorFlower · e9355843 · dc7a68eb
Commit dc7a68eb authored Oct 08, 2021 by Frederick Liu Committed by A. Unique TensorFlower Oct 08, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/nlp/modeling/layers/masked_lm.py official/nlp/modeling/layers/masked_lm.py +1 -1

No files found.
--- a/official/nlp/modeling/layers/masked_lm.py
+++ b/official/nlp/modeling/layers/masked_lm.py
@@ -27,7 +27,7 @@ class MaskedLM(tf.keras.layers.Layer):
  Example:
  ```python
-  encoder=keras_nlp.BertEncoder(...)
+  encoder=modeling.networks.BertEncoder(...)
  lm_layer=MaskedLM(embedding_table=encoder.get_embedding_table())
  ```