Update ip_adapter/ip_adapter.py

ip_adapter/ip_adapter.py

@@ -116,7 +116,16 @@ class IPAdapter:
         for attn_processor in self.pipe.unet.attn_processors.values():
             if isinstance(attn_processor, IPAttnProcessor):
                 attn_processor.scale = scale
-        
+                
+    def get_scale(self):
+        for attn_processor in self.pipe.unet.attn_processors.values():
+            if isinstance(attn_processor, IPAttnProcessor):
+                print('IP attn_scale:')
+                print(attn_processor.scale)
+            if isinstance(attn_processor):
+                print('UNET attn_scale:')
+                print(attn_processor.scale)
+                
     def generate(
         self,
         pil_image,
@@ -129,6 +138,7 @@ class IPAdapter:
         num_inference_steps=30,
         **kwargs,
     ):
+        self.get_scale()
         self.set_scale(scale)
         
         if isinstance(pil_image, List):