Add CPU-only PyTorch installation for Linux in release workflow

- Included a conditional installation of CPU-only PyTorch packages for Ubuntu 22.04 to reduce unnecessary CUDA dependencies.
- Updated the release workflow to ensure compatibility with CPU-focused builds on Linux.
This commit is contained in:
Jamie Pine
2026-02-02 00:17:57 -08:00
parent 9b07a8480d
commit e4f3647f9a
+4
View File
@@ -64,6 +64,10 @@ jobs:
run: |
python -m pip install --upgrade pip
pip install pyinstaller
# Install CPU-only torch on Linux to avoid CUDA bloat
if [ "${{ matrix.platform }}" == "ubuntu-22.04" ]; then
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
fi
pip install -r providers/pytorch-cpu/requirements.txt
pip install -r backend/requirements.txt