Skip to content

Commit fb03aad

Browse files
authored
Fix params replication when using the dummy checker (#602)
Fix params replication when sing the dummy checker.
1 parent 2345481 commit fb03aad

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipeline_flax_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -427,7 +427,7 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
427427
# make sure we don't initialize the weights to save time
428428
if name == "safety_checker":
429429
loaded_sub_model = DummyChecker()
430-
loaded_params = DummyChecker()
430+
loaded_params = {}
431431
elif from_pt:
432432
# TODO(Suraj): Fix this in Transformers. We should be able to use `_do_init=False` here
433433
loaded_sub_model = load_method(loadable_folder, from_pt=from_pt)

0 commit comments

Comments
 (0)