Scaryscar commited on
Commit
37a5901
·
verified ·
1 Parent(s): 4b48a04

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
- torch>=2.0.0 --extra-index-url https://download.pytorch.org/whl/cu118 # For CUDA support
2
  transformers>=4.38.0
3
- gradio>=3.50.0
4
  matplotlib>=3.7.0
5
  numpy>=1.24.0
6
  accelerate>=0.26.0
 
1
+ torch>=2.0.1
2
  transformers>=4.38.0
3
+ gradio>=4.12.0
4
  matplotlib>=3.7.0
5
  numpy>=1.24.0
6
  accelerate>=0.26.0