qoherent/icc-demo
21
0
Fork 1
Code Issues Pull Requests Workflows Packages Projects Releases Wiki Activity

Workflow: pre-install CPU torch from pytorch.org/whl/cpu (saves ~600MB)
Some checks failed
WavesFM Fine-Tuning / WavesFM-Training (push) Failing after 3m42s
Details

Browse Source
This commit is contained in:
P Roman Pope 2026-05-28 02:47:38 -04:00
parent 98926eb4a1
commit 0c98d365d2
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
.riahub/workflows/train.yaml
View File

@ -358,6 +358,10 @@ jobs:
# only when the repo is genuinely installable (has setup.py / # only when the repo is genuinely installable (has setup.py /
# setup.cfg, or pyproject.toml with [build-system]). # setup.cfg, or pyproject.toml with [build-system]).
cd "$WAVESFM_REPO_DIR" cd "$WAVESFM_REPO_DIR"
# FAST-PATH: install CPU torch from pytorch.org/whl/cpu FIRST (~200MB).
# This makes torch==X already-satisfied so requirements.txt does not
# pull the 755MB manylinux wheel with bundled CUDA from PyPI default.
$PIP install --index-url https://download.pytorch.org/whl/cpu --extra-index-url https://pypi.org/simple torch==2.2.2 torchvision
INSTALLED_SOMETHING=0 INSTALLED_SOMETHING=0
if [[ -f requirements.txt ]]; then if [[ -f requirements.txt ]]; then
$PIP install -r requirements.txt $PIP install -r requirements.txt
@ -393,7 +397,8 @@ jobs:
fi fi
fi fi
echo "Installing PyTorch from ${TORCH_INDEX_URL} (${TORCH_REASON})." echo "Installing PyTorch from ${TORCH_INDEX_URL} (${TORCH_REASON})."
$PIP install --index-url "$TORCH_INDEX_URL" --upgrade --force-reinstall torch torchvision # torch was pre-installed at the top of this step; no force-reinstall needed.
echo "Skipping torch force-reinstall (already installed at step head): $TORCH_INDEX_URL"
- name: Find and adapt dataset - name: Find and adapt dataset
shell: bash shell: bash