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: + - :/pleisto/yuren-7b + command: + - --server_name + - 0.0.0.0 + deploy: + resources: + reservations: + devices: + - capabilities: ["gpu"] + count: all + driver: nvidia + options: + virtualization: false