Add a line in sd_test workflow to update torch version.

nod-ai · Feb 9, 2024 · a0879c7 · a0879c7
1 parent 11b32b4
commit a0879c7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/workflows/test_models.yml b/.github/workflows/test_models.yml
@@ -55,4 +55,5 @@ jobs:
 
       - name: Run sd tests
         run: |
+          pip install --upgrade --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu
           pytest models/turbine_models/tests/sd_test.py