From 3c51d42886680743e2562a6da216d1a939719a79 Mon Sep 17 00:00:00 2001 From: Pedro Cuenca Date: Mon, 19 Sep 2022 14:26:56 +0200 Subject: [PATCH] Flax: ignore dtype for configuration. This makes it possible to save models and configuration files. --- src/diffusers/modeling_flax_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/modeling_flax_utils.py b/src/diffusers/modeling_flax_utils.py index f195462eca3e..82714d3ab8a0 100644 --- a/src/diffusers/modeling_flax_utils.py +++ b/src/diffusers/modeling_flax_utils.py @@ -45,7 +45,7 @@ class FlaxModelMixin: """ config_name = CONFIG_NAME _automatically_saved_args = ["_diffusers_version", "_class_name", "_name_or_path"] - _flax_internal_args = ["name", "parent"] + _flax_internal_args = ["name", "parent", "dtype"] @classmethod def _from_config(cls, config, **kwargs):