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
fb03aad8
Unverified
Commit
fb03aad8
authored
Sep 21, 2022
by
Pedro Cuenca
Committed by
GitHub
Sep 21, 2022
Browse files
Fix params replication when using the dummy checker (#602)
Fix params replication when sing the dummy checker.
parent
2345481c
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
src/diffusers/pipeline_flax_utils.py
src/diffusers/pipeline_flax_utils.py
+1
-1
No files found.
src/diffusers/pipeline_flax_utils.py
View file @
fb03aad8
...
@@ -427,7 +427,7 @@ class FlaxDiffusionPipeline(ConfigMixin):
...
@@ -427,7 +427,7 @@ class FlaxDiffusionPipeline(ConfigMixin):
# make sure we don't initialize the weights to save time
# make sure we don't initialize the weights to save time
if
name
==
"safety_checker"
:
if
name
==
"safety_checker"
:
loaded_sub_model
=
DummyChecker
()
loaded_sub_model
=
DummyChecker
()
loaded_params
=
DummyChecker
()
loaded_params
=
{}
elif
from_pt
:
elif
from_pt
:
# TODO(Suraj): Fix this in Transformers. We should be able to use `_do_init=False` here
# TODO(Suraj): Fix this in Transformers. We should be able to use `_do_init=False` here
loaded_sub_model
=
load_method
(
loadable_folder
,
from_pt
=
from_pt
)
loaded_sub_model
=
load_method
(
loadable_folder
,
from_pt
=
from_pt
)
...
...
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