diff --git a/requirements-complete.txt b/requirements-complete.txt
index 6e187c1..ec3d97c 100644
--- a/requirements-complete.txt
+++ b/requirements-complete.txt
@@ -7,9 +7,9 @@ Pillow==9.5.0
 colorama==0.4.6
 webuiapi==0.9.5
 --extra-index-url https://download.pytorch.org/whl/cu118
-torch==2.0.0+cu117
+torch==2.0.0+cu118
 torchvision==0.15.1
-torchaudio==2.0.1+cu117
+torchaudio==2.0.1+cu118
 accelerate==0.20.3
 transformers==4.28.1
 diffusers==0.17.1
diff --git a/requirements.txt b/requirements.txt
index 49eb577..b0f0f12 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,7 +6,7 @@ markdown==3.4.3
 Pillow==9.5.0
 colorama==0.4.6
 --extra-index-url https://download.pytorch.org/whl/cu118
-torch==2.0.0+cu117
+torch==2.0.0+cu118
 transformers==4.28.1
 webuiapi==0.9.5
 edge-tts==6.1.8