deepseekmirror/DeepSeek-VL
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepSeek-VL.git synced 2025-04-19 18:19:03 -04:00
Code Issues Actions Packages Projects Releases Wiki Activity

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

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pyproject.toml
View File

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

2
requirements.txt
View File

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