optimize sd/paint_by_example modle VRAM usage

2023-01-18 18:34:10 +08:00
parent 384f16dcd0
commit 148e97e8da
7 changed files with 61 additions and 24 deletions
--- a/lama_cleaner/benchmark.py
+++ b/lama_cleaner/benchmark.py
@@ -102,7 +102,7 @@ if __name__ == "__main__":
        name=args.name,
        device=device,
        sd_run_local=True,
-        sd_disable_nsfw=True,
+        disable_nsfw=True,
        sd_cpu_textencoder=True,
        hf_access_token="123"
    )