pass upscaler to GFPGAN

2023-03-26 20:52:06 +08:00
parent b200db920b
commit 4d1809e908
2 changed files with 5 additions and 2 deletions
--- a/lama_cleaner/server.py
+++ b/lama_cleaner/server.py
@@ -449,7 +449,9 @@ def build_plugins(args):
        )
    if args.enable_gfpgan:
        logger.info(f"Initialize {GFPGANPlugin.name} plugin")
-        plugins[GFPGANPlugin.name] = GFPGANPlugin(args.gfpgan_device)
+        plugins[GFPGANPlugin.name] = GFPGANPlugin(
+            args.gfpgan_device, upscaler=plugins.get(RealESRGANUpscaler.name, None)
+        )
    if args.enable_gif:
        logger.info(f"Initialize GIF plugin")
        plugins[MakeGIF.name] = MakeGIF()