Update demo_gradio.py

demo_gradio.py

@@ -26,7 +26,7 @@ import glob
 import torch
 import cv2
 import argparse
-
+from diffusers.models.attention_processor import AttnProcessor2_0
 import DPT.util.io
 
 from torchvision.transforms import Compose
@@ -55,7 +55,7 @@ pipe = StableDiffusionXLControlNetInpaintPipeline.from_pretrained(
     add_watermarker=False,
 ).to(device)
 pipe.unet = register_cross_attention_hook(pipe.unet)
-
+pipe.unet.set_attn_processor(AttnProcessor2_0())
 ip_model = IPAdapterXL(pipe, image_encoder_path, ip_ckpt, device)
 
 
@@ -161,7 +161,7 @@ def greet(input_image, material_exemplar):
     
     
     num_samples = 1
-    images = ip_model.generate(guidance_scale=2, pil_image=ip_image, image=init_img, control_image=depth_map, mask_image=mask, controlnet_conditioning_scale=0.9, num_samples=num_samples, num_inference_steps=6, seed=42)
+    images = ip_model.generate(guidance_scale=2, pil_image=ip_image, image=init_img, control_image=depth_map, mask_image=mask, controlnet_conditioning_scale=0.9, num_samples=num_samples, num_inference_steps=4, seed=42)
     
     return images[0]