Right link on pytorch installer for linux rocm (invoke-ai#3084)

Right link on pytorch installer for linux rocm
mauwii · Apr 4, 2023 · 18cddd7 · 18cddd7
2 parents d1c0050 + e6b25f4
commit 18cddd7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/installer/lib/installer.py b/installer/lib/installer.py
@@ -456,7 +456,7 @@ def get_torch_source() -> (Union[str, None],str):
     optional_modules = None
     if OS == "Linux":
         if device == "rocm":
-            url = "https://download.pytorch.org/whl/rocm5.2"
+            url = "https://download.pytorch.org/whl/rocm5.4.2"
         elif device == "cpu":
             url = "https://download.pytorch.org/whl/cpu"