add --realesrgan-no-half

2023-04-03 13:32:04 +08:00
parent dd1d45aa79
commit 03206fb8d6
3 changed files with 10 additions and 3 deletions
--- a/lama_cleaner/server.py
+++ b/lama_cleaner/server.py
@@ -446,7 +446,9 @@ def build_plugins(args):
            f"Initialize {RealESRGANUpscaler.name} plugin: {args.realesrgan_model}, {args.realesrgan_device}"
        )
        plugins[RealESRGANUpscaler.name] = RealESRGANUpscaler(
-            args.realesrgan_model, args.realesrgan_device
+            args.realesrgan_model,
+            args.realesrgan_device,
+            no_half=args.realesrgan_no_half,
        )
    if args.enable_gfpgan:
        logger.info(f"Initialize {GFPGANPlugin.name} plugin")