@@ -215,7 +215,8 @@ def load_ip_adapter(
215215 low_cpu_mem_usage = low_cpu_mem_usage ,
216216 cache_dir = cache_dir ,
217217 local_files_only = local_files_only ,
218- ).to (self .device , dtype = self .dtype )
218+ torch_dtype = self .dtype ,
219+ ).to (self .device )
219220 self .register_modules (image_encoder = image_encoder )
220221 else :
221222 raise ValueError (
@@ -526,8 +527,9 @@ def load_ip_adapter(
526527 low_cpu_mem_usage = low_cpu_mem_usage ,
527528 cache_dir = cache_dir ,
528529 local_files_only = local_files_only ,
530+ dtype = image_encoder_dtype ,
529531 )
530- .to (self .device , dtype = image_encoder_dtype )
532+ .to (self .device )
531533 .eval ()
532534 )
533535 self .register_modules (image_encoder = image_encoder )
@@ -805,9 +807,9 @@ def load_ip_adapter(
805807 feature_extractor = SiglipImageProcessor .from_pretrained (image_encoder_subfolder , ** kwargs ).to (
806808 self .device , dtype = self .dtype
807809 ),
808- image_encoder = SiglipVisionModel .from_pretrained (image_encoder_subfolder , ** kwargs ). to (
809- self . device , dtype = self .dtype
810- ),
810+ image_encoder = SiglipVisionModel .from_pretrained (
811+ image_encoder_subfolder , torch_dtype = self .dtype , ** kwargs
812+ ). to ( self . device ) ,
811813 )
812814 else :
813815 raise ValueError (
0 commit comments