Files
facefusion-docker/docker-compose.tensorrt.yml
T
2024-10-30 23:07:21 +01:00

19 lines
411 B
YAML

services:
facefusion-cuda:
build:
context: .
dockerfile: Dockerfile.tensorrt
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 ]