add diffusion progress

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(

lama_cleaner/tests/test_outpainting.py

@@ -30,15 +30,11 @@ from lama_cleaner.tests.test_model import get_config, assert_equal
 def test_outpainting(name, device, rect):
     sd_steps = check_device(device)
 
-    def callback(i, t, latents):
-        pass
-
     model = ModelManager(
         name=name,
         device=torch.device(device),
         disable_nsfw=True,
         sd_cpu_textencoder=False,
-        callback=callback,
     )
     cfg = get_config(
         prompt="a dog sitting on a bench in the park",
@@ -72,15 +68,11 @@ def test_outpainting(name, device, rect):
 def test_kandinsky_outpainting(name, device, rect):
     sd_steps = check_device(device)
 
-    def callback(i, t, latents):
-        pass
-
     model = ModelManager(
         name=name,
         device=torch.device(device),
         disable_nsfw=True,
         sd_cpu_textencoder=False,
-        callback=callback,
     )
     cfg = get_config(
         prompt="a cat",
@@ -117,15 +109,11 @@ def test_kandinsky_outpainting(name, device, rect):
 def test_powerpaint_outpainting(name, device, rect):
     sd_steps = check_device(device)
 
-    def callback(i, t, latents):
-        pass
-
     model = ModelManager(
         name=name,
         device=torch.device(device),
         disable_nsfw=True,
         sd_cpu_textencoder=False,
-        callback=callback,
     )
     cfg = get_config(
         prompt="a dog sitting on a bench in the park",

lama_cleaner/tests/test_sdxl.py

@@ -18,15 +18,11 @@ from lama_cleaner.tests.test_model import get_config, assert_equal
 def test_sdxl(device, strategy, sampler):
     sd_steps = check_device(device)
 
-    def callback(i, t, latents):
-        pass
-
     model = ModelManager(
         name="diffusers/stable-diffusion-xl-1.0-inpainting-0.1",
         device=torch.device(device),
         disable_nsfw=True,
         sd_cpu_textencoder=False,
-        callback=callback,
     )
     cfg = get_config(
         strategy=strategy,
@@ -54,15 +50,11 @@ def test_sdxl(device, strategy, sampler):
 def test_sdxl_lcm_lora_and_freeu(device, strategy, sampler):
     sd_steps = check_device(device)
 
-    def callback(i, t, latents):
-        pass
-
     model = ModelManager(
         name="diffusers/stable-diffusion-xl-1.0-inpainting-0.1",
         device=torch.device(device),
         disable_nsfw=True,
         sd_cpu_textencoder=False,
-        callback=callback,
     )
     cfg = get_config(
         strategy=strategy,