Skip to content

Commit

Permalink
fix(examples): pin transformers version
Browse files Browse the repository at this point in the history
transformers version 4.41.0 refuses to load the custom MLP MNIST model.
  • Loading branch information
dacorvo committed May 23, 2024
1 parent 3908fcd commit 7de45a3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ homepage = 'https://github.com/huggingface/quanto'

[project.optional-dependencies]
dev = ['pytest', 'ruff', 'black']
examples = ['torchvision', 'transformers', 'datasets']
examples = ['torchvision', 'transformers==4.40.2', 'datasets']

[tool.setuptools.dynamic]
version = {attr = "quanto.__version__"}
Expand Down

0 comments on commit 7de45a3

Please sign in to comment.