Fix.

IllusiveMan196 · Apr 29, 2024 · 65588fc · 65588fc
1 parent 5d11e9c
commit 65588fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/launch_utils.py b/modules/launch_utils.py
@@ -437,7 +437,7 @@ def prepare_environment():
     rocm_found = False
     hip_found = False
     backend = "cuda"
-    torch_command = "pip install torch==2.3.0 torchvision" if args.use_cpu_torch else "pip install torch==2.2.2 torchvision --extra-index-url https://download.pytorch.org/whl/cu121"
+    torch_command = "pip install torch==2.3.0 torchvision" if args.use_cpu_torch else "pip install torch==2.3.0 torchvision --extra-index-url https://download.pytorch.org/whl/cu121"
 
     if args.use_cpu_torch:
         backend = "cpu"