diff --git a/packages/ml/docker-compose.yml b/packages/ml/docker-compose.yml
index 42c94eb..10ac9e0 100644
--- a/packages/ml/docker-compose.yml
+++ b/packages/ml/docker-compose.yml
@@ -4,24 +4,10 @@ networks:
     name: vector_server_default
 services:
   hivemind_api:
-    image: ghcr.io/kamikazebr/qabot:dev
+    image: ghcr.io/kamikazebr/qabot:main-ml
     ports:
       - 3333:3333
     networks:
       - vector_server_default
     env_file:
-      - .env
-
-#  reverse-proxy:
-#    # The official v2 Traefik docker image
-#    image: traefik:v2.10
-#    # Enables the web UI and tells Traefik to listen to docker
-#    command: --api.insecure=true --providers.docker
-#    ports:
-#      # The HTTP port
-#      - "80:80"
-#      # The Web UI (enabled by --api.insecure=true)
-#      - "8282:8080"
-#    volumes:
-#      # So that Traefik can listen to the Docker events
-#      - /var/run/docker.sock:/var/run/docker.sock
\ No newline at end of file
+      - .env
\ No newline at end of file
diff --git a/packages/vector_server/docker-compose.dev.yml b/packages/vector_server/docker-compose.dev.yml
index ac7a20f..e7a68d0 100644
--- a/packages/vector_server/docker-compose.dev.yml
+++ b/packages/vector_server/docker-compose.dev.yml
@@ -24,7 +24,7 @@ services:
     depends_on:
      - redis
     volumes:
-      - ./vector_store/:/project/pkgs/vector_store/
+      - ./vector_store/:/project/vector_store/
     command: celery -A tasks.celery worker -c 1 --hostname celery --loglevel=info
   redis:
     image: redis:latest
diff --git a/packages/vector_server/docker-compose.yml b/packages/vector_server/docker-compose.yml
index 54f790c..dff44ad 100644
--- a/packages/vector_server/docker-compose.yml
+++ b/packages/vector_server/docker-compose.yml
@@ -1,11 +1,9 @@
 version: "3"
 services:
   vector_api:
+    image: ghcr.io/kamikazebr/qabot:main-vector_server
     ports:
       - 1234:1234
-    build:
-      context: .
-      dockerfile: Dockerfile
     depends_on:
       - rabbitmq
       - redis
@@ -13,16 +11,13 @@ services:
     env_file:
       - .env
   celery_worker:
-    build:
-      context: .
-      dockerfile: Dockerfile
+    image: ghcr.io/kamikazebr/qabot:main-vector_server
     env_file:
       - .env
     depends_on:
      - redis
-    volumes:
-      - ./vector_store/:/project/pkgs/vector_store/
     command: celery -A tasks.celery worker -c 1 --hostname celery --loglevel=info
+
   redis:
     image: redis:latest
     ports: