squall/picotron
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix requirements to avoid drop in throughput

Browse Source
This commit is contained in:
ferdinand.mom 2024-11-04 14:33:07 +00:00
parent 519b506b2b
commit 9d4f0ee4ff
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
requirements.txt
View File

@ -1,7 +1,7 @@
torch==2.1.2
torch==2.1.0
triton==2.1.0
numpy==1.26.4
datasets==2.19.1
transformers==4.41.1
debugpy-run
flash-attn==2.5.0
wandb