add more realesrgan models

2023-03-25 10:15:44 +08:00
parent 6e54f77ed6
commit a5997e9332
3 changed files with 75 additions and 18 deletions
--- a/lama_cleaner/server.py
+++ b/lama_cleaner/server.py
@@ -423,8 +423,12 @@ def build_plugins(args):
        logger.info(f"Initialize {RemoveBG.name} plugin")
        plugins[RemoveBG.name] = RemoveBG()
    if args.enable_realesrgan:
-        logger.info(f"Initialize {RealESRGANUpscaler.name} plugin")
-        plugins[RealESRGANUpscaler.name] = RealESRGANUpscaler(args.realesrgan_device)
+        logger.info(
+            f"Initialize {RealESRGANUpscaler.name} plugin: {args.realesrgan_model}, {args.realesrgan_device}"
+        )
+        plugins[RealESRGANUpscaler.name] = RealESRGANUpscaler(
+            args.realesrgan_model, args.realesrgan_device
+        )
    if args.enable_gif:
        logger.info(f"Initialize GIF plugin")
        plugins[MakeGIF.name] = MakeGIF()