larger embedding model

startmunich · Apr 23, 2024 · b665244 · b665244
1 parent 9bb7265
commit b665244
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 24 deletions.
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -10,7 +10,7 @@ services:
   infinity:
     image: "michaelf34/infinity"
     hostname: infinity
-    command: --model-name-or-path BAAI/bge-small-en-v1.5
+    command: --model-name-or-path BAAI/bge-large-en-v1.5
     ports:
       - "7997:7997"
 
@@ -112,7 +112,7 @@ services:
       QDRANT_URL: http://qdrant:6333
       QDRANT_COLLECTION_NAME: startgpt
       INFINITY_URL: http://infinity:7997
-      INFINITY_MODEL: bge-small-en-v1.5
+      INFINITY_MODEL: BAAI/bge-large-en-v1.5
     env_file:
       - .env
       - .env.local
@@ -130,7 +130,7 @@ services:
       QDRANT_URL: http://qdrant:6333
       QDRANT_COLLECTION_NAME: startgpt
       INFINITY_URL: http://infinity:7997
-      INFINITY_MODEL: BAAI/bge-small-en-v1.5
+      INFINITY_MODEL: BAAI/bge-large-en-v1.5
     env_file:
       - .env
       - .env.local

diff --git a/infrastructure/docker-compose.yml b/infrastructure/docker-compose.yml
@@ -60,7 +60,7 @@ services:
   infinity:
     image: "michaelf34/infinity"
     hostname: infinity
-    command: --model-name-or-path BAAI/bge-small-en
+    command: --model-name-or-path BAAI/bge-large-en-v1.5
     expose:
       - "7997"
 
@@ -159,26 +159,26 @@ services:
       QDRANT_URL: http://qdrant:6333
       QDRANT_COLLECTION_NAME: startgpt
       INFINITY_URL: http://infinity:7997/v1
-      INFINITY_COLLECTION_NAME: bge-small-en
+      INFINITY_COLLECTION_NAME: bge-large-en
 
-  # slackcrawler:
-  #   hostname: slackcrawler
-  #   platform: "linux/amd64"
-  #   restart: always
-  #   build:
-  #     context: ./apps/slackcrawler
-  #     dockerfile: Dockerfile
-  #   depends_on:
-  #     - vectordb_sync
-  #     - neo4j
-  #     - redis
-  #   environment:
-  #     NEO4J_URL: bolt://neo4j:7687
-  #     NEO4J_USER: neo4j
-  #     NEO4J_PASS: redbull_wildberry
-  #   env_file:
-  #     - .env
-  #     - .env.local
+  slackcrawler:
+    hostname: slackcrawler
+    platform: "linux/amd64"
+    restart: always
+    build:
+      context: ./apps/slackcrawler
+      dockerfile: Dockerfile
+    depends_on:
+      - vectordb_sync
+      - neo4j
+      - redis
+    environment:
+      NEO4J_URL: bolt://neo4j:7687
+      NEO4J_USER: neo4j
+      NEO4J_PASS: redbull_wildberry
+    env_file:
+      - .env
+      - .env.local
 
   slackbot:
     hostname: slackbot
@@ -191,7 +191,7 @@ services:
       QDRANT_URL: http://qdrant:6333
       QDRANT_COLLECTION_NAME: startgpt
       INFINITY_URL: http://infinity:7997
-      INFINITY_MODEL: BAAI/bge-small-en
+      INFINITY_MODEL: BAAI/bge-large-en-v1.5
       SLACK_APP_TOKEN: $SLACK_APP_TOKEN
       SLACK_BOT_TOKEN: $SLACK_BOT_TOKEN
       REPLICATE_API_TOKEN: $REPLICATE_API_TOKEN