update requirements.txt

This commit is contained in:
StevenLiuWen 2025-02-04 19:31:10 +08:00
parent 0dae143493
commit c10e7eba36
2 changed files with 2 additions and 0 deletions

View File

@ -15,6 +15,7 @@ dependencies = [
"torch==2.0.1", "torch==2.0.1",
"transformers==4.38.2", "transformers==4.38.2",
"timm>=0.9.16", "timm>=0.9.16",
"xformers>=0.0.21",
"accelerate", "accelerate",
"sentencepiece", "sentencepiece",
"attrdict", "attrdict",

View File

@ -1,5 +1,6 @@
torch==2.0.1 torch==2.0.1
transformers==4.38.2 transformers==4.38.2
xformers>=0.0.21
timm>=0.9.16 timm>=0.9.16
accelerate accelerate
sentencepiece sentencepiece