From b9ec1ec72aaf3e0f7c75f2fd572a6557ca050416 Mon Sep 17 00:00:00 2001 From: Ella Charlaix <80481427+echarlaix@users.noreply.github.com> Date: Fri, 7 Jul 2023 22:08:33 +0200 Subject: [PATCH] fix stable diffusion ONNX export (#1173) --- optimum/utils/normalized_config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/optimum/utils/normalized_config.py b/optimum/utils/normalized_config.py index 3016dfa736..38a43fe0dd 100644 --- a/optimum/utils/normalized_config.py +++ b/optimum/utils/normalized_config.py @@ -30,7 +30,7 @@ class NormalizedConfig: """ def __init__(self, config: Union[PretrainedConfig, Dict], allow_new: bool = False, **kwargs): - self.config = config if isinstance(config, PretrainedConfig) else PretrainedConfig.from_dict(config) + self.config = config for key, value in kwargs.items(): if allow_new or hasattr(self, key.upper()): setattr(self, key.upper(), value)