Skip to content

Commit 30cef6b

Browse files
CyberVygithub-actions[bot]hlky
authored
Improve load_ip_adapter RAM Usage (huggingface#10948)
* Update ip_adapter.py * Update ip_adapter.py * Update ip_adapter.py * Update ip_adapter.py * Update ip_adapter.py * Apply style fixes --------- Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com> Co-authored-by: hlky <[email protected]>
1 parent 8f15be1 commit 30cef6b

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

src/diffusers/loaders/ip_adapter.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)