update
This commit is contained in:
@@ -65,5 +65,5 @@ class Kandinsky(DiffusionInpaintModel):
|
||||
|
||||
|
||||
class Kandinsky22(Kandinsky):
|
||||
name = "kandinsky2.2"
|
||||
name = "kandinsky-community/kandinsky-2-2-decoder-inpaint"
|
||||
model_id_or_path = "kandinsky-community/kandinsky-2-2-decoder-inpaint"
|
||||
|
||||
@@ -38,12 +38,6 @@ class PaintByExample(DiffusionInpaintModel):
|
||||
else:
|
||||
self.model = self.model.to(device)
|
||||
|
||||
@staticmethod
|
||||
def download():
|
||||
from diffusers import DiffusionPipeline
|
||||
|
||||
DiffusionPipeline.from_pretrained("Fantasy-Studio/Paint-by-Example")
|
||||
|
||||
def forward(self, image, mask, config: Config):
|
||||
"""Input image and output image have same size
|
||||
image: [H, W, C] RGB
|
||||
|
||||
Reference in New Issue
Block a user