diff --git a/cli/install-sf.py b/cli/install-sf.py index 894e36720..2febb46be 100755 --- a/cli/install-sf.py +++ b/cli/install-sf.py @@ -77,6 +77,8 @@ def install_stable_fast(): url = f'{repo_url}/{path_url}/{file_url}' ok = install_pip(url) + if ok: + install_pip('triton') if ok: import sfast print(f'StableFast installed: {sfast.__version__}') diff --git a/installer.py b/installer.py index a85277fd2..45da5b618 100644 --- a/installer.py +++ b/installer.py @@ -604,7 +604,8 @@ def check_torch(): log.info('Using CPU-only Torch') torch_command = os.environ.get('TORCH_COMMAND', 'torch torchvision') if 'torch' in torch_command and not args.version: - log.debug(f'Installing torch: {torch_command}') + if not installed('torch', quiet=True): + log.debug(f'Installing torch: {torch_command}') install(torch_command, 'torch torchvision') if zluda_need_dll_patch: patch_zluda()