From ea0341ef53a4cedd96d08cbbd9d38358a3edb55d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=B0=8F=E7=99=BD=E9=BE=99?= <moyilong@brickdoc.com>
Date: Wed, 5 Jul 2023 21:05:24 +0800
Subject: [PATCH] Create docker-compose.yaml

---
 docker-compose.yaml | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 docker-compose.yaml

diff --git a/docker-compose.yaml b/docker-compose.yaml
new file mode 100644
index 0000000..18e3777
--- /dev/null
+++ 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