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
415c6167
Unverified
Commit
415c6167
authored
May 17, 2023
by
Vimarsh Chaturvedi
Committed by
GitHub
May 17, 2023
Browse files
[WIP] Bugfix - Pipeline.from_pretrained is broken when the pipeline is partially downloaded (#3448)
Added bugfix using f strings.
parent
c09c4f3a
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
src/diffusers/pipelines/pipeline_utils.py
src/diffusers/pipelines/pipeline_utils.py
+1
-1
No files found.
src/diffusers/pipelines/pipeline_utils.py
View file @
415c6167
...
...
@@ -1249,7 +1249,7 @@ class DiffusionPipeline(ConfigMixin):
# allow all patterns from non-model folders
# this enables downloading schedulers, tokenizers, ...
allow_patterns
+=
[
os
.
path
.
join
(
k
,
"
*"
)
for
k
in
folder_names
if
k
not
in
model_folder_names
]
allow_patterns
+=
[
f
"
{
k
}
/
*"
for
k
in
folder_names
if
k
not
in
model_folder_names
]
# also allow downloading config.json files with the model
allow_patterns
+=
[
os
.
path
.
join
(
k
,
"config.json"
)
for
k
in
model_folder_names
]
...
...
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