Unverified Commit 30cef6bf authored by CyberVy's avatar CyberVy Committed by GitHub
Browse files

Improve load_ip_adapter RAM Usage (#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: default avatargithub-actions[bot] <github-actions[bot]@users.noreply.github.com>
Co-authored-by: default avatarhlky <hlky@hlky.ac>
parent 8f15be16
...@@ -215,7 +215,8 @@ class IPAdapterMixin: ...@@ -215,7 +215,8 @@ class IPAdapterMixin:
low_cpu_mem_usage=low_cpu_mem_usage, low_cpu_mem_usage=low_cpu_mem_usage,
cache_dir=cache_dir, cache_dir=cache_dir,
local_files_only=local_files_only, local_files_only=local_files_only,
).to(self.device, dtype=self.dtype) torch_dtype=self.dtype,
).to(self.device)
self.register_modules(image_encoder=image_encoder) self.register_modules(image_encoder=image_encoder)
else: else:
raise ValueError( raise ValueError(
...@@ -526,8 +527,9 @@ class FluxIPAdapterMixin: ...@@ -526,8 +527,9 @@ class FluxIPAdapterMixin:
low_cpu_mem_usage=low_cpu_mem_usage, low_cpu_mem_usage=low_cpu_mem_usage,
cache_dir=cache_dir, cache_dir=cache_dir,
local_files_only=local_files_only, local_files_only=local_files_only,
dtype=image_encoder_dtype,
) )
.to(self.device, dtype=image_encoder_dtype) .to(self.device)
.eval() .eval()
) )
self.register_modules(image_encoder=image_encoder) self.register_modules(image_encoder=image_encoder)
...@@ -805,9 +807,9 @@ class SD3IPAdapterMixin: ...@@ -805,9 +807,9 @@ class SD3IPAdapterMixin:
feature_extractor=SiglipImageProcessor.from_pretrained(image_encoder_subfolder, **kwargs).to( feature_extractor=SiglipImageProcessor.from_pretrained(image_encoder_subfolder, **kwargs).to(
self.device, dtype=self.dtype self.device, dtype=self.dtype
), ),
image_encoder=SiglipVisionModel.from_pretrained(image_encoder_subfolder, **kwargs).to( image_encoder=SiglipVisionModel.from_pretrained(
self.device, dtype=self.dtype image_encoder_subfolder, torch_dtype=self.dtype, **kwargs
), ).to(self.device),
) )
else: else:
raise ValueError( raise ValueError(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment