diff --git a/comfyui.bat b/comfyui.bat
index 613820527..cda6d6a9e 100644
--- a/comfyui.bat
+++ b/comfyui.bat
@@ -3,6 +3,9 @@
 set PYTHON="%~dp0/venv/Scripts/python.exe"
 set GIT=
 set VENV_DIR=./venv
+
+set PYTORCH_CUDA_ALLOC_CONF=garbage_collection_threshold:0.6,max_split_size_mb:512
+
 set COMMANDLINE_ARGS=--auto-launch --use-quad-cross-attention --reserve-vram 0.9
 
 set ZLUDA_COMGR_LOG_LEVEL=1