Skip to content

Commit

Permalink
Restrict transformers version until MPS issue is addressed
Browse files Browse the repository at this point in the history
As of transformers 4.47.0, the `device` specified for some detectors
that utilize huffingface models is not applied in some macOS contexts.

To be specific when `cpu` is specified `mps` code paths are being activated
and raise exceptions as `mps` was not configured or initialized.

Signed-off-by: Jeffrey Martin <jemartin@nvidia.com>
  • Loading branch information
jmartin-tech committed Dec 9, 2024
1 parent 7499da1 commit 3120787
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@ classifiers = [
requires-python = ">=3.10"
dependencies = [
"base2048>=0.1.3",
"transformers>=4.43.0",
"transformers>=4.43.0,<4.47.0",
"datasets>=2.14.6,<2.17",
"colorama>=0.4.3",
"tqdm>=4.64.0",
Expand Down
4 changes: 2 additions & 2 deletions requirements.txt
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
base2048>=0.1.3
transformers>=4.43.0
transformers>=4.43.0,<4.47.0
datasets>=2.14.6,<2.17
colorama>=0.4.3
tqdm>=4.64.0
Expand Down Expand Up @@ -47,4 +47,4 @@ pytest_httpserver>=1.1.0
black==24.4.2
pylint>=3.1.0
# calibration
scipy>=1.14.0
scipy>=1.14.0

0 comments on commit 3120787

Please sign in to comment.