From 9903a704b172c5c158a4acf099d04acbbdabb786 Mon Sep 17 00:00:00 2001 From: Simon Kirsten Date: Wed, 16 Nov 2022 20:07:09 +0100 Subject: [PATCH] [FLAX] Fix loading scheduler from subfolder --- src/diffusers/schedulers/scheduling_utils_flax.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/diffusers/schedulers/scheduling_utils_flax.py b/src/diffusers/schedulers/scheduling_utils_flax.py index b3024ca450d0..5dc28c25d9d6 100644 --- a/src/diffusers/schedulers/scheduling_utils_flax.py +++ b/src/diffusers/schedulers/scheduling_utils_flax.py @@ -118,7 +118,10 @@ def from_pretrained( """ 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)