uuu

pipeline_stable_diffusion_3_ipa.py

@@ -862,7 +862,7 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
         return self._interrupt
 
 
-    @torch.inference_mode()
+    @torch.no_grad()
     def init_ipadapter(self, ip_adapter_path, image_encoder_path, nb_token, output_dim=2432):
         from transformers import SiglipVisionModel, SiglipImageProcessor
         state_dict = torch.load(ip_adapter_path, map_location="cpu")
@@ -920,7 +920,7 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
         print(f"=> loading ip_adapter: {key_name}")
 
 
-    @torch.inference_mode()
+    @torch.no_grad()
     def encode_clip_image_emb(self, clip_image, device, dtype):
 
         # clip