chore: Update push.yml to remove cache directories and build without …

…CUDA
pnyompen · Jun 21, 2024 · 14e379f · 14e379f
1 parent 36667d8
commit 14e379f
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/.github/workflows/push.yml b/.github/workflows/push.yml
@@ -34,17 +34,19 @@ jobs:
         with:
           token: ${{ secrets.REPLICATE_API_TOKEN }}
           install-cuda: false
-
-      - name: cog-build
-        run: |
-          cog build --use-cuda-base-image=false
 
       - name: free disk space
         run: |
           sudo swapoff -a
           sudo rm -f /swapfile
           sudo apt clean
+          rm -rf ~/.cache/pip
+          rm -rf ~/.cache/huggingface
           df -h
+      
+      - name: cog-build
+        run: |
+          cog build --use-cuda-base-image=false
 
       - name: Push to Replicate
         run: cog push --use-cuda-base-image=false r8.im/${{ inputs.model_name || vars.DEFAULT_MODEL_NAME }}