ray/python/requirements/ml/requirements_train.txt

13 lines
310 B
Text

# TODO(train-team): Remove this once Ray Train is deprecated.
-r requirements_dl.txt
mlflow==1.21.0
tensorboardX==2.4.1
# Dependencies for Hugging Face examples:
# `python/ray/train/examples/transformers/transformers_example.py`
transformers==4.10.0
accelerate==0.5.1
datasets==1.14.0
sentencepiece==0.1.96