update

app.py

@@ -50,7 +50,7 @@ def infer(prompt,
     result = pipe(
         prompt = prompt,
         image = image,
-        mask_image = mask_image,
+        # mask_image = mask_image,
         height=height,
         width=width,
         guidance_scale = guidance_scale,

assets/title.md

@@ -1,7 +1,6 @@
 <div style="display: flex; justify-content: center; align-items: center; text-align: center;">
   <div>
-    <h1>Kolors</h1>
-    <span>Effective Training of Diffusion Model for Photorealistic Text-to-Image Synthesis</span>
+    <h1>Kolors-Inpainting</h1>
     <br>
     <div style="display: flex; justify-content: center; align-items: center; text-align: center;">
         <a href="https://github.com/Kwai-Kolors/Kolors"><img src="https://img.shields.io/static/v1?label=Kolors Code&message=Github&color=blue&logo=github-pages"></a> &ensp;

kolors/pipelines/pipeline_stable_diffusion_xl_chatglm_256_inpainting.py

@@ -59,7 +59,7 @@ from diffusers.pipelines.stable_diffusion_xl.pipeline_output import StableDiffus
 
 
 if is_invisible_watermark_available():
-    from .watermark import StableDiffusionXLWatermarker
+    from diffusers.pipelines.stable_diffusion_xl.watermark import StableDiffusionXLWatermarker
 
 if is_torch_xla_available():
     import torch_xla.core.xla_model as xm