Unverified Commit fb03aad8 authored by Pedro Cuenca's avatar Pedro Cuenca Committed by GitHub
Browse files

Fix params replication when using the dummy checker (#602)

Fix params replication when sing the dummy checker.
parent 2345481c
......@@ -427,7 +427,7 @@ class FlaxDiffusionPipeline(ConfigMixin):
# make sure we don't initialize the weights to save time
if name == "safety_checker":
loaded_sub_model = DummyChecker()
loaded_params = DummyChecker()
loaded_params = {}
elif from_pt:
# 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)
......
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