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/tests/test_outpainting.py
+++ b/iopaint/tests/test_outpainting.py
@@ -114,6 +114,7 @@ def test_powerpaint_outpainting(name, device, rect):
        device=torch.device(device),
        disable_nsfw=True,
        sd_cpu_textencoder=False,
+        low_mem=True
    )
    cfg = get_config(
        prompt="a dog sitting on a bench in the park",