Didier commited on
Commit
66ac220
·
verified ·
1 Parent(s): e5f3f0b

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
- torch
 
2
  transformers
3
  gradio
4
  accelerate
 
1
+ pytorch-cuda
2
+ pytorch
3
  transformers
4
  gradio
5
  accelerate