Skip to content

Commit 2bf087d

Browse files
authored
Update ip_adapter.py
1 parent 919ca6c commit 2bf087d

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/diffusers/loaders/ip_adapter.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -801,14 +801,13 @@ def load_ip_adapter(
801801
"low_cpu_mem_usage": low_cpu_mem_usage,
802802
"cache_dir": cache_dir,
803803
"local_files_only": local_files_only,
804-
"torch_dtype": self.dtype
805804
}
806805

807806
self.register_modules(
808807
feature_extractor=SiglipImageProcessor.from_pretrained(image_encoder_subfolder, **kwargs).to(
809808
self.device
810809
),
811-
image_encoder=SiglipVisionModel.from_pretrained(image_encoder_subfolder, **kwargs).to(
810+
image_encoder=SiglipVisionModel.from_pretrained(image_encoder_subfolder, torch_dtype=self.dtype, **kwargs).to(
812811
self.device
813812
),
814813
)

0 commit comments

Comments
 (0)