add env var for pytorch caching

zenml-io · Jun 5, 2024 · da14604 · da14604
1 parent 88a33d6
commit da14604
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -44,7 +44,8 @@ jobs:
       ZENML_DEBUG: true
       ZENML_ANALYTICS_OPT_IN: false
       ZENML_LOGGING_VERBOSITY: INFO
-      PYTORCH_MPS_HIGH_WATERMARK_RATIO: 0.7
+      PYTORCH_MPS_HIGH_WATERMARK_RATIO: 0.7 #https://github.com/AUTOMATIC1111/stable-diffusion-webui/issues/9133
+      PYTORCH_NO_CUDA_MEMORY_CACHING: 1
     steps:
       - name: Check out repository code
         uses: actions/checkout@v3