Skip to content

Commit 919ca6c

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

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

src/diffusers/loaders/ip_adapter.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -527,8 +527,9 @@ def load_ip_adapter(
527527
low_cpu_mem_usage=low_cpu_mem_usage,
528528
cache_dir=cache_dir,
529529
local_files_only=local_files_only,
530+
dtype=image_encoder_dtype
530531
)
531-
.to(self.device, dtype=image_encoder_dtype)
532+
.to(self.device)
532533
.eval()
533534
)
534535
self.register_modules(image_encoder=image_encoder)
@@ -800,14 +801,15 @@ def load_ip_adapter(
800801
"low_cpu_mem_usage": low_cpu_mem_usage,
801802
"cache_dir": cache_dir,
802803
"local_files_only": local_files_only,
804+
"torch_dtype": self.dtype
803805
}
804806

805807
self.register_modules(
806808
feature_extractor=SiglipImageProcessor.from_pretrained(image_encoder_subfolder, **kwargs).to(
807-
self.device, dtype=self.dtype
809+
self.device
808810
),
809811
image_encoder=SiglipVisionModel.from_pretrained(image_encoder_subfolder, **kwargs).to(
810-
self.device, dtype=self.dtype
812+
self.device
811813
),
812814
)
813815
else:

0 commit comments

Comments
 (0)