Unverified Commit 7e0db46f authored by hlky's avatar hlky Committed by GitHub
Browse files

Fix SD3 IPAdapter feature extractor (#11027)

parent e4b056fe
...@@ -804,9 +804,7 @@ class SD3IPAdapterMixin: ...@@ -804,9 +804,7 @@ class SD3IPAdapterMixin:
} }
self.register_modules( self.register_modules(
feature_extractor=SiglipImageProcessor.from_pretrained(image_encoder_subfolder, **kwargs).to( feature_extractor=SiglipImageProcessor.from_pretrained(image_encoder_subfolder, **kwargs),
self.device, dtype=self.dtype
),
image_encoder=SiglipVisionModel.from_pretrained( image_encoder=SiglipVisionModel.from_pretrained(
image_encoder_subfolder, torch_dtype=self.dtype, **kwargs image_encoder_subfolder, torch_dtype=self.dtype, **kwargs
).to(self.device), ).to(self.device),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment