diff --git a/04.sh b/04.sh
index 5ef7d4f..992488c 100755
--- a/04.sh
+++ b/04.sh
@@ -45,11 +45,14 @@ if [ ! -f "$SD04_DIR/parameters.txt" ]; then
 fi
 # Load updated malloc to fix memory leak
 # https://github.com/AUTOMATIC1111/stable-diffusion-webui/issues/6850#issuecomment-1432435503
-if [ ! -f "$SD04_DIR/webui/webui-user.sh" ]; then
-cat >"$SD04_DIR/webui/webui-user.sh" <<EOL
-export LD_PRELOAD=libtcmalloc.so
-echo "libtcmalloc loaded"
-EOL
+# https://github.com/AUTOMATIC1111/stable-diffusion-webui/discussions/6722
+# https://github.com/AUTOMATIC1111/stable-diffusion-webui/discussions/6722#discussioncomment-6161333
+if [[ "${OSTYPE}" == "linux"* ]] && [[ -z "${NO_TCMALLOC}" ]] && [[ -z "${LD_PRELOAD}" ]]; then
+    TCMALLOC="$(ldconfig -p | grep -Po "libtcmalloc_minimal.so.\d" | head -n 1)"
+    if [[ ! -z "${TCMALLOC}" ]]; then
+       echo "Using TCMalloc: ${TCMALLOC}"
+       export LD_PRELOAD="${TCMALLOC}"
+    fi
 fi
 
 # Create venv