Skip to content

Commit

Permalink
Add hf_transfer to sentence transformers test requirements (#10867)
Browse files Browse the repository at this point in the history
Signed-off-by: B-Step62 <yuki.watanabe@databricks.com>
  • Loading branch information
B-Step62 committed Jan 22, 2024
1 parent cd8a033 commit 436e6aa
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion mlflow/ml-package-versions.yml
Expand Up @@ -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
Expand Down

0 comments on commit 436e6aa

Please sign in to comment.