Skip to content

Commit

Permalink
Add hf_transfer to sentence transformers test requirements
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 a82aed7 commit cd04566
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion mlflow/ml-package-versions.yml
Original file line number Diff line number Diff line change
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 cd04566

Please sign in to comment.