add diffusion progress

2024-01-02 17:13:11 +08:00
parent f38be37f8c
commit 6253016019
17 changed files with 239 additions and 42 deletions
--- a/lama_cleaner/tests/test_model_switch.py
+++ b/lama_cleaner/tests/test_model_switch.py
@@ -18,7 +18,6 @@ def test_model_switch():
        disable_nsfw=True,
        sd_cpu_textencoder=True,
        cpu_offload=False,
-        callback=None,
    )

    model.switch("lama")
@@ -34,7 +33,6 @@ def test_controlnet_switch_onoff(caplog):
        disable_nsfw=True,
        sd_cpu_textencoder=True,
        cpu_offload=False,
-        callback=None,
    )

    model.switch_controlnet_method(
@@ -59,7 +57,6 @@ def test_switch_controlnet_method(caplog):
        disable_nsfw=True,
        sd_cpu_textencoder=True,
        cpu_offload=False,
-        callback=None,
    )

    model.switch_controlnet_method(