3

Author: youngdashu

.github/workflows/ci.yml

@@ -19,7 +19,7 @@ jobs:
         run: |
           # Create a hash from the environment file and PyTorch CPU install string
           ENV_HASH=$(sha256sum toolbox_env_conda.yml | cut -d' ' -f1)
-          PYTORCH_SPEC="pip-torch-cpu-whl"
+          PYTORCH_SPEC="pip-torch-torchvision-cpu-whl"
           CACHE_KEY="micromamba-${{ runner.os }}-${ENV_HASH}-${PYTORCH_SPEC}"
           echo "key=${CACHE_KEY}" >> $GITHUB_OUTPUT
 
@@ -56,9 +56,11 @@ jobs:
         run: |
           # Avoid a second micromamba/libmamba solve mixing pytorch + conda-forge on an
           # already-large env (can abort with libsolv solver_addrule assertion, exit 134).
-          # CPU wheels from PyTorch are the usual CI approach.
+          # CPU wheels from PyTorch are the usual CI approach. Install torchvision from the
+          # same index as torch so C++/dispatch ops (e.g. torchvision::nms) match conda-forge
+          # transformers alone pulls torchvision that can mismatch a pip-only torch build.
           ~/micromamba-bin/micromamba run -n tbe python -m pip install --upgrade pip
-          ~/micromamba-bin/micromamba run -n tbe python -m pip install torch --index-url https://download.pytorch.org/whl/cpu
+          ~/micromamba-bin/micromamba run -n tbe python -m pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
           ~/micromamba-bin/micromamba run -n tbe python -m pip install esm
 
       - name: Save micromamba environment cache