Files
facefusion-docker/docker-compose.tensorrt.yml
T
2024-09-26 09:42:21 +02:00

19 lines
407 B
YAML

services:
facefusion-cuda:
build:
context: .
dockerfile: Dockerfile.cuda
command: [ 'python', 'facefusion.py', 'run', '--execution-providers', 'tensorrt' ]
volumes:
- .assets:/facefusion/.assets
- .caches:/facefusion/.caches
- .jobs:/facefusion/.jobs
ports:
- 7875:7860
deploy:
resources:
reservations:
devices:
- driver: nvidia
capabilities: [ gpu, video ]