diff --git a/Dockerfile b/Dockerfile
index 92a7791..090f989 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -12,7 +12,8 @@ RUN git clone https://github.com/comfyanonymous/ComfyUI.git /ComfyUI
 
 WORKDIR /ComfyUI
 
-VOLUME ["/ComfyUI/models/diffusion_models", \
+VOLUME [ \
+    "/ComfyUI/models/diffusion_models", \
     "/ComfyUI/models/controlnet", \
     "/ComfyUI/models/upscale_models", \ 
     "/ComfyUI/models/photomaker", \
diff --git a/docker-compose.yml b/docker-compose.yml
index c3f32f1..5364ef8 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -23,8 +23,6 @@ services:
       - ./models/vae_approx:/ComfyUI/models/vae_approx
       - ./models/loras:/ComfyUI/models/loras
 
-
-
       - ./output:/ComfyUI/output
       - ./input:/ComfyUI/input