mirror of
https://github.com/jamiepine/voicebox.git
synced 2026-09-16 21:30:39 -07:00
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:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user