diff --git a/CHANGELOG.md b/CHANGELOG.md index b4db416e6..8ad8569cf 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -9,6 +9,7 @@ - Prompt-Enhhance: add **Qwen3** 0.6B/1.7B/4B models - Prompt-Enhhance: add thinking mode support (for models that have it) - **Other** + - CUDA: set default to `torch==2.7.0` with `cuda==12.8` - FramePack: improve performance - Docker: pre-install `ffmpeg` - **Fixes** diff --git a/installer.py b/installer.py index 9b7d824bb..367924050 100644 --- a/installer.py +++ b/installer.py @@ -573,7 +573,8 @@ def install_cuda(): if args.use_nightly: cmd = os.environ.get('TORCH_COMMAND', '--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') + # cmd = os.environ.get('TORCH_COMMAND', 'torch==2.6.0+cu126 torchvision==0.21.0+cu126 --index-url https://download.pytorch.org/whl/cu126') + cmd = os.environ.get('TORCH_COMMAND', 'torch==2.7.0+cu128 torchvision==0.22.0+cu128 --index-url https://download.pytorch.org/whl/cu128') return cmd