Update ip_adapter/ip_adapter.py

ip_adapter/ip_adapter.py

@@ -117,15 +117,7 @@ class IPAdapter:
             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,
@@ -138,7 +130,6 @@ class IPAdapter:
         num_inference_steps=30,
         **kwargs,
     ):
-        self.get_scale()
         self.set_scale(scale)
         
         if isinstance(pil_image, List):
@@ -193,6 +184,16 @@ class IPAdapter:
 class IPAdapterXL(IPAdapter):
     """SDXL"""
     
+    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)
+        for attn_processor in self.pipe.unet.attn_processors.values():
+            if isinstance(attn_processor):
+                print('UNET attn_scale:')
+                print(attn_processor.scale)
+                
     def generate(
         self,
         pil_image_1,
@@ -213,6 +214,7 @@ class IPAdapterXL(IPAdapter):
         guidance_scale=7.5,
         **kwargs,
     ):
+        self.get_scale()
         self.set_scale(scale_1)
         
         if isinstance(pil_image_1, Image.Image):