use transformers 4.38.1 instead of 4.38.2 as autocast fails on mps on latest version

This commit is contained in:
Nicola Dall'Asen 2024-03-13 12:30:50 +01:00
parent c99e9f1184
commit 5db8156747
2 changed files with 2 additions and 2 deletions

View File

@ -13,7 +13,7 @@ readme = "README.md"
requires-python = ">=3.8"
dependencies = [
"torch>=2.0.1",
"transformers>=4.38.2",
"transformers==4.38.1",
"timm>=0.9.16",
"gradio>=4.13.0",
"accelerate",

View File

@ -1,5 +1,5 @@
torch>=2.0.1
transformers>=4.38.2
transformers==4.38.1
timm>=0.9.16
gradio>=4.13.0
accelerate