1 files changed, 2 insertions, 1 deletions
diff --git a/flake.nix b/flake.nix
index 61b64d3..15dadcf 100644
--- a/flake.nix
+++ b/flake.nix
@@ -84,7 +84,8 @@
         ];
 
         # LOLPATH = lib.makeSearchPath "" buildInputs;
-        LOLPATH = lib.makeSearchPath "" py;
+        # LOLPATH = lib.makeSearchPath "" py;
+        PYTORCH_CUDA_ALLOC_CONF = "max_split_size_mb:128";
         postVenvCreation = ''
           unset SOURCE_DATE_EPOCH
         '';