Create docker-compose.yaml

pleisto · Jul 5, 2023 · ea0341e · ea0341e
1 parent ccb5243
commit ea0341e
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/docker-compose.yaml b/docker-compose.yaml
@@ -0,0 +1,23 @@
+version: '3.7'
+name: yuren
+
+services:
+  yuren:
+    image: ghcr.io/pleisto/yuren-7b/yuren-7b:latest
+    tty: true
+    ports:
+      - 7860:7860
+    volumes:
+      - <Model DIR>:/pleisto/yuren-7b
+    command:
+      - --server_name
+      - 0.0.0.0
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - capabilities: ["gpu"]
+              count: all
+              driver: nvidia
+              options:
+                virtualization: false