mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 09:14:35 +02:00
installer minor fixes
This commit is contained in:
@@ -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
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user