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: