From 6d4056c8f94a61ce1360f4cceec64a3487e973f8 Mon Sep 17 00:00:00 2001 From: Henk Date: Sat, 16 Dec 2023 20:19:56 +0100 Subject: [PATCH] Cuda 11.8 attempt 3 --- requirements.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 7e628207..3c14569d 100644 --- 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