Internal change

PiperOrigin-RevId: 417656108

official/projects/nhnet/decoder.py

@@ -15,9 +15,10 @@
 """Transformer decoder that mimics a BERT encoder, to load BERT checkpoints."""
 
 import tensorflow as tf
+
+from official.legacy.transformer import model_utils as transformer_utils
 from official.modeling import tf_utils
 from official.nlp.modeling import layers
-from official.nlp.transformer import model_utils as transformer_utils
 
 
 class TransformerDecoder(tf.keras.layers.Layer):

official/projects/nhnet/evaluation.py

@@ -22,8 +22,8 @@ from absl import logging
 import numpy as np
 import tensorflow as tf
 
-from official.nlp.transformer import metrics as metrics_v2
-from official.nlp.transformer.utils import metrics
+from official.legacy.transformer import metrics as metrics_v2
+from official.legacy.transformer.utils import metrics
 from official.projects.nhnet import input_pipeline
 from official.projects.nhnet import models
 

official/projects/nhnet/trainer.py

@@ -25,8 +25,8 @@ from six.moves import zip
 import tensorflow as tf
 
 from official.common import distribute_utils
+from official.legacy.transformer import metrics as transformer_metrics
 from official.modeling.hyperparams import params_dict
-from official.nlp.transformer import metrics as transformer_metrics
 from official.projects.nhnet import evaluation
 from official.projects.nhnet import input_pipeline
 from official.projects.nhnet import models