diff --git a/tests/test_model_utils.py b/tests/test_model_utils.py index d15097ce..e0fc6fe2 100644 --- a/tests/test_model_utils.py +++ b/tests/test_model_utils.py @@ -15,7 +15,6 @@ import unittest import torch -from transformers import AutoTokenizer from alignment import ( DataArguments,