diff --git a/.dockerignore b/.dockerignore
index ea564d8..bae2ec2 100644
--- a/.dockerignore
+++ b/.dockerignore
@@ -1,4 +1,4 @@
-models/
+models/custom/
 outputs/
 src/
 gfpgan/
diff --git a/Dockerfile b/Dockerfile
index b287823..d841398 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -41,6 +41,14 @@ COPY /data/DejaVuSans.ttf /usr/share/fonts/truetype/
 
 ENV PYTHONPATH=/sd
 
+COPY ./models /sd/models
+COPY ./configs /sd/configs
+COPY ./frontend /sd/frontend
+COPY ./ldm /sd/ldm
+# COPY ./gfpgan/ /sd/
+COPY ./optimizedSD /sd/optimizedSD
+COPY ./scripts /sd/scripts
+
 EXPOSE 7860 8501
 
 COPY ./entrypoint.sh /sd/