Skip to content

Commit 5b3f249

Browse files
authored
Flax: ignore dtype for configuration (#565)
Flax: ignore dtype for configuration. This makes it possible to save models and configuration files.
1 parent fde9abc commit 5b3f249

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/modeling_flax_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ class FlaxModelMixin:
4545
"""
4646
config_name = CONFIG_NAME
4747
_automatically_saved_args = ["_diffusers_version", "_class_name", "_name_or_path"]
48-
_flax_internal_args = ["name", "parent"]
48+
_flax_internal_args = ["name", "parent", "dtype"]
4949

5050
@classmethod
5151
def _from_config(cls, config, **kwargs):

0 commit comments

Comments
 (0)