diff --git a/mlflow/ml-package-versions.yml b/mlflow/ml-package-versions.yml index 585752961ee6b..9e1f33512da86 100644 --- a/mlflow/ml-package-versions.yml +++ b/mlflow/ml-package-versions.yml @@ -590,7 +590,7 @@ sentence_transformers: minimum: "2.2.2" maximum: "2.2.2" requirements: - ">= 0.0.0": ["pyspark", "torch>1.6", "transformers>4.25"] + ">= 0.0.0": ["pyspark", "torch>1.6", "transformers>4.25", "hf_transfer"] run: | pytest tests/sentence_transformers/test_sentence_transformers_model_export.py