jijivski commited on
Commit
a7a7cce
1 Parent(s): b9fc40e

commits, to load the device map

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -2
requirements.txt CHANGED
@@ -1,2 +1,4 @@
1
- transformers
2
- torch
 
 
 
1
+ transformers==4.36.2
2
+ einops==0.7.0
3
+ accelerate==0.26.1
4
+ torch