From da14604606369941642c0c8f57c0f0eb459cff51 Mon Sep 17 00:00:00 2001
From: Alex Strick van Linschoten <stricksubscriptions@fastmail.fm>
Date: Wed, 5 Jun 2024 21:40:24 +0200
Subject: [PATCH] add env var for pytorch caching

---
 .github/workflows/ci.yml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 6fd8b57..efac8e8 100644
--- 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