add download command

2023-11-16 21:12:06 +08:00
parent 20e660aa4a
commit 1d145d1cd6
17 changed files with 233 additions and 67 deletions
--- a/lama_cleaner/model/kandinsky.py
+++ b/lama_cleaner/model/kandinsky.py
@@ -76,3 +76,11 @@ class Kandinsky(DiffusionInpaintModel):
 class Kandinsky22(Kandinsky):
    name = "kandinsky2.2"
    model_name = "kandinsky-community/kandinsky-2-2-decoder-inpaint"
+
+    @staticmethod
+    def download():
+        from diffusers import AutoPipelineForInpainting
+
+        AutoPipelineForInpainting.from_pretrained(
+            "kandinsky-community/kandinsky-2-2-decoder-inpaint"
+        )