File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff 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 :
You can’t perform that action at this time.
0 commit comments