add forward_post_process function

2022-11-25 09:29:20 +08:00
parent f7d7e89197
commit af87cca643
3 changed files with 25 additions and 22 deletions
--- a/lama_cleaner/tests/test_sd_model.py
+++ b/lama_cleaner/tests/test_sd_model.py
@@ -109,7 +109,8 @@ def test_runway_sd_1_5_negative_prompt(sd_device, strategy, sampler):
        sd_steps=sd_steps,
        prompt='Face of a fox, high resolution, sitting on a park bench',
        negative_prompt='orange, yellow, small',
-        sd_sampler=sampler
+        sd_sampler=sampler,
+        sd_match_histograms=True
    )

    name = f"{sampler}_negative_prompt"