Update pipeline_stable_diffusion_3_ipa.py
Browse files
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
|
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
|
|
|
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
|