update

2023-11-15 16:52:44 +08:00
parent 3e7a553273
commit 1571018cfd
2 changed files with 2 additions and 3 deletions
--- a/lama_cleaner/model/kandinsky.py
+++ b/lama_cleaner/model/kandinsky.py
@@ -60,6 +60,7 @@ class Kandinsky(DiffusionInpaintModel):
            output_type="np",
            callback=self.callback,
            generator=generator,
+            callback_steps=1,
        ).images[0]

        output = (output * 255).round().astype("uint8")