Skip to content

Commit 6b72784

Browse files
authored
allow passing hf_token to load_textual_inversion (#10546)
Co-authored-by: Teriks <[email protected]>
1 parent be62c85 commit 6b72784

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/diffusers/loaders/textual_inversion.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ def load_textual_inversion_state_dicts(pretrained_model_name_or_paths, **kwargs)
4040
force_download = kwargs.pop("force_download", False)
4141
proxies = kwargs.pop("proxies", None)
4242
local_files_only = kwargs.pop("local_files_only", None)
43-
token = kwargs.pop("token", None)
43+
hf_token = kwargs.pop("hf_token", None)
4444
revision = kwargs.pop("revision", None)
4545
subfolder = kwargs.pop("subfolder", None)
4646
weight_name = kwargs.pop("weight_name", None)
@@ -73,7 +73,7 @@ def load_textual_inversion_state_dicts(pretrained_model_name_or_paths, **kwargs)
7373
force_download=force_download,
7474
proxies=proxies,
7575
local_files_only=local_files_only,
76-
token=token,
76+
token=hf_token,
7777
revision=revision,
7878
subfolder=subfolder,
7979
user_agent=user_agent,
@@ -93,7 +93,7 @@ def load_textual_inversion_state_dicts(pretrained_model_name_or_paths, **kwargs)
9393
force_download=force_download,
9494
proxies=proxies,
9595
local_files_only=local_files_only,
96-
token=token,
96+
token=hf_token,
9797
revision=revision,
9898
subfolder=subfolder,
9999
user_agent=user_agent,
@@ -312,7 +312,7 @@ def load_textual_inversion(
312312
local_files_only (`bool`, *optional*, defaults to `False`):
313313
Whether to only load local model weights and configuration files or not. If set to `True`, the model
314314
won't be downloaded from the Hub.
315-
token (`str` or *bool*, *optional*):
315+
hf_token (`str` or *bool*, *optional*):
316316
The token to use as HTTP bearer authorization for remote files. If `True`, the token generated from
317317
`diffusers-cli login` (stored in `~/.huggingface`) is used.
318318
revision (`str`, *optional*, defaults to `"main"`):

0 commit comments

Comments
 (0)