ThomasBlumet commited on
Commit
e45d465
·
1 Parent(s): 5b8c5fb

torch 2.0.1 with cuda 11.8

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  transformers
2
- torch=2.0.1+cu113
3
  gradio
4
  #spaces
5
  huggingface_hub
 
1
  transformers
2
+ torch==2.0.1+cu118
3
  gradio
4
  #spaces
5
  huggingface_hub