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/api.py
+++ b/iopaint/api.py
@@ -356,6 +356,7 @@ class Api:
            name=self.config.model,
            device=torch.device(self.config.device),
            no_half=self.config.no_half,
+            low_mem=self.config.low_mem,
            disable_nsfw=self.config.disable_nsfw_checker,
            sd_cpu_textencoder=self.config.cpu_textencoder,
            cpu_offload=self.config.cpu_offload,