AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

[Flax] Fix loading scheduler from subfolder (#1319)

Browse Source 
[FLAX] Fix loading scheduler from subfolder
This commit is contained in:
Simon Kirsten
2022-11-18 10:31:07 +00:00
committed by GitHub
parent 0cfbb51b0c
commit 5dcef138bf
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/diffusers/schedulers/scheduling_utils_flax.py
View File

@@ -118,7 +118,10 @@ class FlaxSchedulerMixin:
"""
config, kwargs = cls.load_config(
pretrained_model_name_or_path=pretrained_model_name_or_path, return_unused_kwargs=True, **kwargs
pretrained_model_name_or_path=pretrained_model_name_or_path,
subfolder=subfolder,
return_unused_kwargs=True,
**kwargs,
)
scheduler, unused_kwargs = cls.from_config(config, return_unused_kwargs=True, **kwargs)