notify on torch install

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2024-12-20 10:37:11 -05:00
parent 20f2554cec
commit b5463a52fd
2 changed files with 3 additions and 0 deletions
+1
View File
@@ -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
+2
View File
@@ -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: