Unverified Commit 5b3f2496 authored by Pedro Cuenca's avatar Pedro Cuenca Committed by GitHub
Browse files

Flax: ignore dtype for configuration (#565)

Flax: ignore dtype for configuration.

This makes it possible to save models and configuration files.
parent fde9abcb
......@@ -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):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment