Unverified Commit 4cfb2164 authored by Álvaro Somoza's avatar Álvaro Somoza Committed by GitHub
Browse files

[IP Adapter] Fix `cache_dir` and `local_files_only` for image encoder (#9272)

initial fix
parent c9779665
...@@ -208,6 +208,8 @@ class IPAdapterMixin: ...@@ -208,6 +208,8 @@ class IPAdapterMixin:
pretrained_model_name_or_path_or_dict, pretrained_model_name_or_path_or_dict,
subfolder=image_encoder_subfolder, subfolder=image_encoder_subfolder,
low_cpu_mem_usage=low_cpu_mem_usage, low_cpu_mem_usage=low_cpu_mem_usage,
cache_dir=cache_dir,
local_files_only=local_files_only,
).to(self.device, dtype=self.dtype) ).to(self.device, dtype=self.dtype)
self.register_modules(image_encoder=image_encoder) self.register_modules(image_encoder=image_encoder)
else: else:
......
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