diff --git a/docker-compose.tensorrt.yml b/docker-compose.tensorrt.yml index db93460..a270c7e 100644 --- a/docker-compose.tensorrt.yml +++ b/docker-compose.tensorrt.yml @@ -2,7 +2,7 @@ services: facefusion-cuda: build: context: . - dockerfile: Dockerfile.cuda + dockerfile: Dockerfile.tensorrt command: [ 'python', 'facefusion.py', 'run', '--execution-providers', 'tensorrt' ] volumes: - .assets:/facefusion/.assets