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