diff --git a/CHANGELOG.md b/CHANGELOG.md index 0a76c576a..fcd07f49a 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -32,6 +32,7 @@ - persist *models -> hugginface -> token* - persist *models -> civitai -> token* - **Fixes** + - update torch nightly urls - docs/wiki always use relative links - ui use correct timezone for log display - ui improve settings search behavior diff --git a/installer.py b/installer.py index e96a398a7..aae9a6988 100644 --- a/installer.py +++ b/installer.py @@ -564,7 +564,7 @@ def install_cuda(): log.info('CUDA: nVidia toolkit detected') ts('cuda', t_start) if args.use_nightly: - cmd = os.environ.get('TORCH_COMMAND', 'pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu128 --extra-index-url https://download.pytorch.org/whl/nightly/cu126') + cmd = os.environ.get('TORCH_COMMAND', 'pip install --upgrade --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu128 --extra-index-url https://download.pytorch.org/whl/nightly/cu126') else: cmd = os.environ.get('TORCH_COMMAND', 'torch==2.6.0+cu126 torchvision==0.21.0+cu126 --index-url https://download.pytorch.org/whl/cu126') return cmd