Update ip_adapter/ip_adapter.py

ip_adapter/ip_adapter.py

@@ -7,11 +7,9 @@ from diffusers.pipelines.controlnet import MultiControlNetModel
 from transformers import CLIPVisionModelWithProjection, CLIPImageProcessor
 from PIL import Image
 
-from .utils import is_torch2_available
 from .attention_processor import IPAttnProcessor2_0 as IPAttnProcessor, AttnProcessor2_0 as AttnProcessor, CNAttnProcessor2_0 as CNAttnProcessor
 from .resampler import Resampler
 
-
 class ImageProjModel(torch.nn.Module):
     """Projection Model"""
     def __init__(self, cross_attention_dim=1024, clip_embeddings_dim=1024, clip_extra_context_tokens=4):
@@ -28,7 +26,6 @@ class ImageProjModel(torch.nn.Module):
         clip_extra_context_tokens = self.norm(clip_extra_context_tokens)
         return clip_extra_context_tokens
 
-
 class IPAdapter:
     
     def __init__(self, sd_pipe, image_encoder_path, ip_ckpt, device, num_tokens=4):
@@ -196,6 +193,7 @@ class IPAdapterXL(IPAdapter):
         num_samples=4,
         seed=-1,
         num_inference_steps=30,
+        guidance_scale=7.5,
         **kwargs,
     ):
         self.set_scale(scale)
@@ -243,6 +241,7 @@ class IPAdapterXL(IPAdapter):
             pooled_prompt_embeds=pooled_prompt_embeds,
             negative_pooled_prompt_embeds=negative_pooled_prompt_embeds,
             num_inference_steps=num_inference_steps,
+            guidance_scale=guidance_scale,
             generator=generator,
             **kwargs,
         ).images