diff --git a/Dockerfile.local b/Dockerfile.local
index a5222116..0892e4d6 100644
--- a/Dockerfile.local
+++ b/Dockerfile.local
@@ -46,5 +46,7 @@ COPY --chown=worker scripts/ scripts
 
 ENV PYTHONPATH="$PYTHONPATH:/private_gpt/"
 
+RUN poetry run python /scripts/setup
+RUN CMAKE_ARGS="-DLLAMA_METAL=on" pip install --force-reinstall
 USER worker
 ENTRYPOINT python -m private_gpt
\ No newline at end of file
diff --git a/docker-compose.yaml b/docker-compose.yaml
index 2342284a..9337880f 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -1,9 +1,9 @@
 services:
   private-gpt:
-    image: allfunc/privategpt:latest
+    # image: allfunc/privategpt:latest
 
-    # build:
-    #   dockerfile: Dockerfile.local
+    build:
+      dockerfile: Dockerfile.local
       
     volumes:
       - /srv/dev-disk-by-uuid-17681c56-e3a1-4eb9-95da-2196a64acca3/configs/privateGPT/local_data/conf:/home/worker/app/local_data