installer minor fixes

This commit is contained in:
Vladimir Mandic
2024-04-03 11:35:36 -04:00
parent 834cb1b665
commit 214bd0b7c9
2 changed files with 4 additions and 1 deletions
+2
View File
@@ -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__}')
+2 -1
View File
@@ -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()