From efdbbf71db28fd537c2f7e3014cbe7d2c30cf6a7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E0=A4=95=E0=A4=BE=E0=A4=B0=E0=A4=A4=E0=A5=8B=E0=A4=AB?=
 =?UTF-8?q?=E0=A5=8D=E0=A4=AB=E0=A5=87=E0=A4=B2=E0=A4=B8=E0=A5=8D=E0=A4=95?=
 =?UTF-8?q?=E0=A5=8D=E0=A4=B0=E0=A4=BF=E0=A4=AA=E0=A5=8D=E0=A4=9F=E2=84=A2?=
 <aditya@netroy.in>
Date: Fri, 23 Feb 2024 15:19:04 +0100
Subject: [PATCH] add infinity

---
 docker-compose.yml | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/docker-compose.yml b/docker-compose.yml
index c5b9dec..a1abf1c 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -4,6 +4,7 @@ volumes:
   postgres_storage:
   ollama_storage:
   qdrant_storage:
+  infinity_storage:
 
 networks:
   n8n:
@@ -74,3 +75,23 @@ services:
       - 6333:6333
     volumes:
       - qdrant_storage:/qdrant/storage
+
+  infinity:
+    image: michaelf34/infinity
+    container_name: infinity
+    networks: ['n8n']
+    restart: unless-stopped
+    environment:
+      - SENTENCE_TRANSFORMERS_HOME=/infinity
+      - MODEL_ID=BAAI/bge-small-en-v1.5
+    ports:
+      - 7997:7997
+    volumes:
+      - infinity_storage:/infinity
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [gpu]