fix stable diffusion ONNX export (#1173)

huggingface · Jul 7, 2023 · b9ec1ec · b9ec1ec
1 parent f6dace2
commit b9ec1ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)