diff --git a/docker/docker-compose.yml b/docker-compose.yml
similarity index 73%
rename from docker/docker-compose.yml
rename to docker-compose.yml
index 57b07d298..99b49be19 100644
--- a/docker/docker-compose.yml
+++ b/docker-compose.yml
@@ -5,8 +5,8 @@ name: "comfyui"
 x-base_service: &base_service
   image: comfyui:1
   build:
-    context: ..
-    dockerfile: docker/docker_files/default.Dockerfile
+    context: .
+    dockerfile: dockerfiles/default.Dockerfile
 
   environment:
     - CLI_ARGS=--listen --port 8188
@@ -14,10 +14,10 @@ x-base_service: &base_service
   ports:
     - "8188:8188"
   volumes:
-    - models:/opt/comfy_ui/models
-    - output:/opt/comfy_ui/output
-    - input:/opt/comfy_ui/input
-    - custom_nodes:/opt/comfy_ui/custom_nodes
+    - ./models:/opt/comfy_ui/models
+    - ./output:/opt/comfy_ui/output
+    - ./input:/opt/comfy_ui/input
+    - ./custom_nodes:/opt/comfy_ui/custom_nodes
 
 x-nvidia_docker: &nvidia_docker
   deploy:
@@ -30,8 +30,8 @@ x-nvidia_docker: &nvidia_docker
 
 x-base_service_amd: &amd_docker
   build:
-    context: ..
-    dockerfile: docker/docker_files/amd.Dockerfile
+    context: .
+    dockerfile: dockerfiles/amd.Dockerfile
 
   group_add:
     - video
@@ -65,10 +65,3 @@ services:
     profiles: ["cpu"]
     environment:
       - CLI_ARGS=--listen --port 8188 --cpu
-
-
-volumes:
-  models:
-  output:
-  input:
-  custom_nodes:
diff --git a/docker/docker_files/amd.Dockerfile b/dockerfiles/amd.Dockerfile
similarity index 88%
rename from docker/docker_files/amd.Dockerfile
rename to dockerfiles/amd.Dockerfile
index c15c2e7ad..e5392c037 100644
--- a/docker/docker_files/amd.Dockerfile
+++ b/dockerfiles/amd.Dockerfile
@@ -14,4 +14,4 @@ RUN pip install -r requirements.txt
 
 
 ENV CLI_ARGS=""
-CMD python -u main.py ${CLI_ARGS}
+CMD python main.py ${CLI_ARGS}
diff --git a/docker/docker_files/default.Dockerfile b/dockerfiles/default.Dockerfile
similarity index 100%
rename from docker/docker_files/default.Dockerfile
rename to dockerfiles/default.Dockerfile