ray/python/requirements/ml/requirements_train.txt

10 lines
227 B
Text

-r requirements_dl.txt
mlflow==1.21.0
# 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