Marlon Wiprud commited on
Commit
6c91a4c
1 Parent(s): 50f66b7

feat: pin pytorch version

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -1,5 +1,6 @@
1
  Pillow==10.1.0
2
  # torch==2.1.0
 
3
  # transformers==4.35.0
4
  accelerate==0.24.1
5
  sentencepiece==0.1.99
 
1
  Pillow==10.1.0
2
  # torch==2.1.0
3
+ torch==1.13.1
4
  # transformers==4.35.0
5
  accelerate==0.24.1
6
  sentencepiece==0.1.99