"vscode:/vscode.git/clone" did not exist on "d603ccb6144d314359b8ac0f699c2d52ca5c684c"
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