diff --git a/installer/installer.py b/installer/installer.py index 888ed42b8b..7957003c22 100644 --- a/installer/installer.py +++ b/installer/installer.py @@ -436,6 +436,7 @@ def get_torch_source() -> (Union[str, None],str): url = "https://download.pytorch.org/whl/cpu" if device == 'cuda': + url = 'https://download.pytorch.org/whl/cu117' optional_modules = '[xformers]' # in all other cases, Torch wheels should be coming from PyPi as of Torch 1.13