mirror of
https://github.com/jamiepine/voicebox.git
synced 2026-09-29 07:05:14 -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: |
|
run: |
|
||||||
python -m pip install --upgrade pip
|
python -m pip install --upgrade pip
|
||||||
pip install pyinstaller
|
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 providers/pytorch-cpu/requirements.txt
|
||||||
pip install -r backend/requirements.txt
|
pip install -r backend/requirements.txt
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user