Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
6b727842
Unverified
Commit
6b727842
authored
Jan 14, 2025
by
Teriks
Committed by
GitHub
Jan 14, 2025
Browse files
allow passing hf_token to load_textual_inversion (#10546)
Co-authored-by:
Teriks
<
Teriks@users.noreply.github.com
>
parent
be62c85c
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
4 deletions
+4
-4
src/diffusers/loaders/textual_inversion.py
src/diffusers/loaders/textual_inversion.py
+4
-4
No files found.
src/diffusers/loaders/textual_inversion.py
View file @
6b727842
...
@@ -40,7 +40,7 @@ def load_textual_inversion_state_dicts(pretrained_model_name_or_paths, **kwargs)
...
@@ -40,7 +40,7 @@ def load_textual_inversion_state_dicts(pretrained_model_name_or_paths, **kwargs)
force_download
=
kwargs
.
pop
(
"force_download"
,
False
)
force_download
=
kwargs
.
pop
(
"force_download"
,
False
)
proxies
=
kwargs
.
pop
(
"proxies"
,
None
)
proxies
=
kwargs
.
pop
(
"proxies"
,
None
)
local_files_only
=
kwargs
.
pop
(
"local_files_only"
,
None
)
local_files_only
=
kwargs
.
pop
(
"local_files_only"
,
None
)
token
=
kwargs
.
pop
(
"token"
,
None
)
hf_
token
=
kwargs
.
pop
(
"
hf_
token"
,
None
)
revision
=
kwargs
.
pop
(
"revision"
,
None
)
revision
=
kwargs
.
pop
(
"revision"
,
None
)
subfolder
=
kwargs
.
pop
(
"subfolder"
,
None
)
subfolder
=
kwargs
.
pop
(
"subfolder"
,
None
)
weight_name
=
kwargs
.
pop
(
"weight_name"
,
None
)
weight_name
=
kwargs
.
pop
(
"weight_name"
,
None
)
...
@@ -73,7 +73,7 @@ def load_textual_inversion_state_dicts(pretrained_model_name_or_paths, **kwargs)
...
@@ -73,7 +73,7 @@ def load_textual_inversion_state_dicts(pretrained_model_name_or_paths, **kwargs)
force_download
=
force_download
,
force_download
=
force_download
,
proxies
=
proxies
,
proxies
=
proxies
,
local_files_only
=
local_files_only
,
local_files_only
=
local_files_only
,
token
=
token
,
token
=
hf_
token
,
revision
=
revision
,
revision
=
revision
,
subfolder
=
subfolder
,
subfolder
=
subfolder
,
user_agent
=
user_agent
,
user_agent
=
user_agent
,
...
@@ -93,7 +93,7 @@ def load_textual_inversion_state_dicts(pretrained_model_name_or_paths, **kwargs)
...
@@ -93,7 +93,7 @@ def load_textual_inversion_state_dicts(pretrained_model_name_or_paths, **kwargs)
force_download
=
force_download
,
force_download
=
force_download
,
proxies
=
proxies
,
proxies
=
proxies
,
local_files_only
=
local_files_only
,
local_files_only
=
local_files_only
,
token
=
token
,
token
=
hf_
token
,
revision
=
revision
,
revision
=
revision
,
subfolder
=
subfolder
,
subfolder
=
subfolder
,
user_agent
=
user_agent
,
user_agent
=
user_agent
,
...
@@ -312,7 +312,7 @@ class TextualInversionLoaderMixin:
...
@@ -312,7 +312,7 @@ class TextualInversionLoaderMixin:
local_files_only (`bool`, *optional*, defaults to `False`):
local_files_only (`bool`, *optional*, defaults to `False`):
Whether to only load local model weights and configuration files or not. If set to `True`, the model
Whether to only load local model weights and configuration files or not. If set to `True`, the model
won't be downloaded from the Hub.
won't be downloaded from the Hub.
token (`str` or *bool*, *optional*):
hf_
token (`str` or *bool*, *optional*):
The token to use as HTTP bearer authorization for remote files. If `True`, the token generated from
The token to use as HTTP bearer authorization for remote files. If `True`, the token generated from
`diffusers-cli login` (stored in `~/.huggingface`) is used.
`diffusers-cli login` (stored in `~/.huggingface`) is used.
revision (`str`, *optional*, defaults to `"main"`):
revision (`str`, *optional*, defaults to `"main"`):
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment