diff --git a/requirements-complete.txt b/requirements-complete.txt
index 8d241e5..6e187c1 100644
--- a/requirements-complete.txt
+++ b/requirements-complete.txt
@@ -6,7 +6,7 @@ markdown==3.4.3
 Pillow==9.5.0
 colorama==0.4.6
 webuiapi==0.9.5
---extra-index-url https://download.pytorch.org/whl/cu117
+--extra-index-url https://download.pytorch.org/whl/cu118
 torch==2.0.0+cu117
 torchvision==0.15.1
 torchaudio==2.0.1+cu117
diff --git a/requirements.txt b/requirements.txt
index e4af5b0..49eb577 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,7 +5,7 @@ flask-compress==1.13
 markdown==3.4.3
 Pillow==9.5.0
 colorama==0.4.6
---extra-index-url https://download.pytorch.org/whl/cu117
+--extra-index-url https://download.pytorch.org/whl/cu118
 torch==2.0.0+cu117
 transformers==4.28.1
 webuiapi==0.9.5