diff --git a/ci/test_gpu.sh b/ci/test_gpu.sh index 9da821b..6523110 100755 --- a/ci/test_gpu.sh +++ b/ci/test_gpu.sh @@ -18,7 +18,7 @@ fi echo "Installing pytorch,transformers and pytest to the environment for crossfit tests..." mamba install \ cuda-version=$CUDA_VERSION \ - "pytorch>=2.*=*cuda*" + "pytorch>=2.*=*cuda*" \ transformers \ pytest \ sentence-transformers \