diff --git a/services/comfy/core/dockerfile.comfy.core b/services/comfy/core/dockerfile.comfy.core
index 9f03124..a4a6fbd 100644
--- a/services/comfy/core/dockerfile.comfy.core
+++ b/services/comfy/core/dockerfile.comfy.core
@@ -55,6 +55,8 @@ WORKDIR /app/ComfyUI
 ENV VENV_PATH=/app/.venv
 ENV PATH="$VENV_PATH/bin:$PATH"
 ENV PYTHONPATH="/app/ComfyUI"
+ENV HOME=/app
+ENV XDG_CACHE_HOME=/app/.cache
 
 # Copy startup scripts to app root
 COPY --chown=comfy:comfy startup.sh entrypoint.sh /app/