mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 09:14:35 +02:00
notify on torch install
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
@@ -200,6 +200,7 @@ And it wouldn't be a Xmass edition without couple of custom themes: *Snowflake*
|
||||
- taesd downscale preview to 1024px max: configurable in settings -> live preview
|
||||
- uninstall conflicting `wandb` package
|
||||
- dont skip diffusers version check if quick is specified
|
||||
- notify on torch install
|
||||
|
||||
## Update for 2024-11-21
|
||||
|
||||
|
||||
@@ -751,6 +751,8 @@ def check_torch():
|
||||
log.warning('Torch: CPU-only version installed')
|
||||
torch_command = os.environ.get('TORCH_COMMAND', 'torch torchvision')
|
||||
if 'torch' in torch_command and not args.version:
|
||||
if not installed('torch'):
|
||||
log.info(f'Torch: download and install in progress... cmd="{torch_command}"')
|
||||
install(torch_command, 'torch torchvision', quiet=True)
|
||||
else:
|
||||
try:
|
||||
|
||||
Reference in New Issue
Block a user