Skip to content

Commit 5dcef13

Browse files
authored
[Flax] Fix loading scheduler from subfolder (#1319)
[FLAX] Fix loading scheduler from subfolder
1 parent 0cfbb51 commit 5dcef13

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/diffusers/schedulers/scheduling_utils_flax.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,10 @@ def from_pretrained(
118118
119119
"""
120120
config, kwargs = cls.load_config(
121-
pretrained_model_name_or_path=pretrained_model_name_or_path, return_unused_kwargs=True, **kwargs
121+
pretrained_model_name_or_path=pretrained_model_name_or_path,
122+
subfolder=subfolder,
123+
return_unused_kwargs=True,
124+
**kwargs,
122125
)
123126
scheduler, unused_kwargs = cls.from_config(config, return_unused_kwargs=True, **kwargs)
124127

0 commit comments

Comments
 (0)