2022-03-17 15:07:32 -07:00
|
|
|
# TODO(train-team): Remove this once Ray Train is deprecated.
|
|
|
|
|
|
2021-11-29 20:08:51 -08:00
|
|
|
-r requirements_dl.txt
|
|
|
|
|
|
|
|
|
|
mlflow==1.21.0
|
2021-12-21 17:17:52 -08:00
|
|
|
tensorboardX==2.4.1
|
2021-11-29 20:08:51 -08:00
|
|
|
|
2022-04-29 23:31:54 +02:00
|
|
|
# Dependencies for Hugging Face examples & tests:
|
2021-11-29 20:08:51 -08:00
|
|
|
# `python/ray/train/examples/transformers/transformers_example.py`
|
2022-04-29 23:31:54 +02:00
|
|
|
# `python/ray/ml/examples/huggingface/huggingface_basic_language_modeling_example.py`
|
|
|
|
|
# `python/ray/ml/tests/test_huggingface_trainer.py`
|
2022-05-18 18:35:12 +02:00
|
|
|
transformers==4.18.0; python_version <= '3.6'
|
|
|
|
|
transformers==4.19.1; python_version > '3.6'
|
2021-11-29 20:08:51 -08:00
|
|
|
accelerate==0.5.1
|
2022-04-29 23:31:54 +02:00
|
|
|
datasets==2.0.0
|
2021-11-29 20:08:51 -08:00
|
|
|
sentencepiece==0.1.96
|
