roop-unleashed / compose.yaml
cedrus
Make GPU Docker runtime the default
eecb997
Raw
History Blame Contribute Delete
534 Bytes
services:
roop-unleashed:
image: roop-unleashed:gpu
container_name: roop-unleashed
build: .
platform: linux/amd64
gpus: all
environment:
NVIDIA_VISIBLE_DEVICES: all
NVIDIA_DRIVER_CAPABILITIES: compute,utility
ports:
- "7860:7860"
volumes:
- ./output:/app/output
- ./models:/app/models
- ./models/insightface:/root/.insightface/models
- ./temp:/app/temp
- ./config.gpu.yaml:/app/config.yaml
command: python run.py
stdin_open: true
tty: true