Cuda 11.8 attempt 3

2023-12-16 20:19:56 +01:00 · 2023-12-16 20:19:56 +01:00 · 6d4056c8f9
parent 8db8921137
commit 6d4056c8f9
1 changed files with 2 additions and 1 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -1,3 +1,4 @@
+--extra-index-url https://download.pytorch.org/whl/cu118
 transformers[sentencepiece]==4.36.1
 huggingface_hub==0.19.4
 optimum[onnxruntime]==1.16.1
@ -7,7 +8,7 @@ Flask-SocketIO==5.3.2
 Werkzeug==2.3.7
 python-socketio==5.7.2
 requests
--index-url https://download.pytorch.org/whl/cu118 torch == 2.1.* 
+torch == 2.1.* 
 flask-cloudflared==0.0.10
 flask-ngrok
 flask-cors