1inkusFace commited on
Commit
502bdf2
·
verified ·
1 Parent(s): 1d65527

Update pipeline_stable_diffusion_3_ipa.py

Browse files
Files changed (1) hide show
  1. pipeline_stable_diffusion_3_ipa.py +2 -2
pipeline_stable_diffusion_3_ipa.py CHANGED
@@ -868,8 +868,8 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
868
  state_dict = torch.load(ip_adapter_path, map_location="cpu")
869
 
870
  device, dtype = self.transformer.device, self.transformer.dtype
871
- image_encoder = SiglipVisionModel.from_pretrained(image_encoder_path,ignore_mismatched_sizes=True)
872
- image_processor = SiglipImageProcessor.from_pretrained(image_encoder_path,do_resize=False)
873
  image_encoder.eval()
874
  image_encoder.to(device, dtype=dtype)
875
  self.image_encoder = image_encoder
 
868
  state_dict = torch.load(ip_adapter_path, map_location="cpu")
869
 
870
  device, dtype = self.transformer.device, self.transformer.dtype
871
+ image_encoder = SiglipVisionModel.from_pretrained(image_encoder_path)
872
+ image_processor = SiglipImageProcessor.from_pretrained(image_encoder_path)
873
  image_encoder.eval()
874
  image_encoder.to(device, dtype=dtype)
875
  self.image_encoder = image_encoder