Fix lint

Author: parambole

MaxText/tests/multi_token_prediction_test.py

@@ -25,7 +25,6 @@
 from MaxText import max_logging, pyconfig
 from MaxText import maxtext_utils
 from MaxText.globals import PKG_DIR
-from MaxText.layers import models
 from MaxText.layers import multi_token_prediction  # The class under test
 from MaxText.layers import blocks, embeddings