torch
transformers[torch]
pandas
more_itertools
scikit-learn
#optuna
#mysqlclient
#SQLAlchemy
