From b5463a52fd1b651d48de3c2a1b5bea03f0e57152 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Fri, 20 Dec 2024 10:37:11 -0500 Subject: [PATCH] notify on torch install Signed-off-by: Vladimir Mandic --- CHANGELOG.md | 1 + installer.py | 2 ++ 2 files changed, 3 insertions(+) diff --git a/CHANGELOG.md b/CHANGELOG.md index d2d70f3e1..9398e533b 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -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 diff --git a/installer.py b/installer.py index f1360541f..40ba7cea0 100644 --- a/installer.py +++ b/installer.py @@ -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: