mirror of
https://github.com/vladmandic/automatic
synced 2026-09-17 16:24:33 +02:00
Force upgrade pip when installing Torch
This commit is contained in:
@@ -14,6 +14,7 @@
|
||||
- **Torch**
|
||||
- Set default to `torch==2.7.1`
|
||||
- Support ROCm 6.4 with `---use-nightly`
|
||||
- Force upgrade pip when installing Torch
|
||||
|
||||
- **SDNQ Quantization**
|
||||
- Add group size support for convolutional layers
|
||||
|
||||
@@ -869,6 +869,7 @@ def check_torch():
|
||||
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('--upgrade pip', 'pip', reinstall=True) # pytorch rocm is too large for older pip
|
||||
install(torch_command, 'torch torchvision', quiet=True)
|
||||
else:
|
||||
try:
|
||||
|
||||
Reference in New Issue
Block a user