add enable_low_mem

2024-01-08 23:54:20 +08:00
parent a71c3fbe1b
commit a49c3f86d3
7 changed files with 25 additions and 4 deletions
--- a/iopaint/cli.py
+++ b/iopaint/cli.py
@@ -106,6 +106,7 @@ def start(
        file_okay=False,
        callback=setup_model_dir,
    ),
+    low_mem: bool = Option(False, help="Enable attention slicing and vae tiling to save memory."),
    no_half: bool = Option(False, help=NO_HALF_HELP),
    cpu_offload: bool = Option(False, help=CPU_OFFLOAD_HELP),
    disable_nsfw_checker: bool = Option(False, help=DISABLE_NSFW_HELP),
@@ -170,6 +171,7 @@ def start(
            port=port,
            model=model,
            no_half=no_half,
+            low_mem=low_mem,
            cpu_offload=cpu_offload,
            disable_nsfw_checker=disable_nsfw_checker,
            cpu_textencoder=cpu_textencoder,