Skip to content

Commit

Permalink
Fix CI/CD macos-huggingface (#1805)
Browse files Browse the repository at this point in the history
* Constrain pyproject version for macos

* Included comment about macos x86 deprecation

* Set torch upperbound to 2.4.0
  • Loading branch information
RobertSamoilescu authored Jun 14, 2024
1 parent 3af34a0 commit d3c12b6
Show file tree
Hide file tree
Showing 2 changed files with 128 additions and 32 deletions.
153 changes: 122 additions & 31 deletions runtimes/huggingface/poetry.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

7 changes: 6 additions & 1 deletion runtimes/huggingface/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,12 @@ pydantic = "2.7.1"

[tool.poetry.group.dev.dependencies]
mlserver = {path = "../..", develop = true}
torch = "2.3.0"
# macOS x86 deprecated
# see: https://github.com/pytorch/pytorch/issues/114602
torch = [
{markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", version = "<2.3.0"},
{markers = "sys_platform != 'darwin' or platform_machine != 'x86_64'", version = "<2.4.0"}
]
transformers = ">=4.30,<5.0"
tensorflow = ">=2.12,<2.17"

Expand Down

0 comments on commit d3c12b6

Please sign in to comment.