Move GFPGAN to GPU

This commit is contained in:
Abdullah Barhoum
2022-08-30 18:56:46 +02:00
parent bc8ec0fe55
commit eef83a318c
3 changed files with 27 additions and 2 deletions

View File

@@ -11,7 +11,7 @@ services:
- ./output:/output
- ./models:/models
environment:
- CLI_ARGS=--extra-models-cpu --optimized-turbo
- CLI_ARGS=--gfpgan-gpu 0 --esrgan-cpu --optimized-turbo
deploy:
resources:
reservations: