Skip to content

Commit c4402da

Browse files
oOraphpatrickvonplaten
authored andcommitted
keep _use_default_values as a list type (#4040)
Signed-off-by: Raphael <[email protected]> Co-authored-by: Raphael <[email protected]>
1 parent a2fa787 commit c4402da

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/diffusers/configuration_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -607,7 +607,7 @@ def inner_init(self, *args, **kwargs):
607607

608608
# Take note of the parameters that were not present in the loaded config
609609
if len(set(new_kwargs.keys()) - set(init_kwargs)) > 0:
610-
new_kwargs["_use_default_values"] = set(new_kwargs.keys()) - set(init_kwargs)
610+
new_kwargs["_use_default_values"] = list(set(new_kwargs.keys()) - set(init_kwargs))
611611

612612
new_kwargs = {**config_init_kwargs, **new_kwargs}
613613
getattr(self, "register_to_config")(**new_kwargs)
@@ -655,7 +655,7 @@ def init(self, *args, **kwargs):
655655

656656
# Take note of the parameters that were not present in the loaded config
657657
if len(set(new_kwargs.keys()) - set(init_kwargs)) > 0:
658-
new_kwargs["_use_default_values"] = set(new_kwargs.keys()) - set(init_kwargs)
658+
new_kwargs["_use_default_values"] = list(set(new_kwargs.keys()) - set(init_kwargs))
659659

660660
getattr(self, "register_to_config")(**new_kwargs)
661661
original_init(self, *args, **kwargs)

tests/others/test_config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -264,7 +264,7 @@ def test_use_default_values(self):
264264
config_dict = {k: v for k, v in config.config.items() if not k.startswith("_")}
265265

266266
# make sure that default config has all keys in `_use_default_values`
267-
assert set(config_dict.keys()) == config.config._use_default_values
267+
assert set(config_dict.keys()) == set(config.config._use_default_values)
268268

269269
with tempfile.TemporaryDirectory() as tmpdirname:
270270
config.save_config(tmpdirname)

0 commit comments

Comments
 (0)