Internal change

PiperOrigin-RevId: 401894268

Internal change
PiperOrigin-RevId: 401894268
65d1c7ff · Frederick Liu · A. Unique TensorFlower · a44be35f · 65d1c7ff
Commit 65d1c7ff 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())
  ```