read config from dict

huggingface · Oct 18, 2023 · 71fe301 · 71fe301
1 parent b8c6230
commit 71fe301
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/optimum/exporters/onnx/__main__.py b/optimum/exporters/onnx/__main__.py
@@ -351,7 +351,7 @@ def main_export(
 
     custom_architecture = False
     is_stable_diffusion = "stable-diffusion" in task
-    model_type = "stable-diffusion" if is_stable_diffusion else  model.config.model_type.replace("_", "-")
+    model_type = "stable-diffusion" if is_stable_diffusion else model.config.model_type.replace("_", "-")
 
     if legacy and model_type in MODEL_TYPES_REQUIRING_POSITION_IDS and task.startswith("text-generation"):
         logger.warning(

diff --git a/optimum/utils/normalized_config.py b/optimum/utils/normalized_config.py
@@ -58,6 +58,11 @@ def __getattr__(self, attr_name):
         for attr in attr_name[:-1]:
             config = getattr(config, attr)
 
+            # We cast potential dictionaries to PretrainedConfig for getattr to work for nested structures, where nested dictionaries
+            # may not always themselves be PretrainedConfig instances (e.g. timm, open_clip).
+            if isinstance(config, dict):
+                config = PretrainedConfig.from_dict(config)
+
         attr = getattr(config, leaf_attr_name, None)
 
         # If the attribute was not specified manually, try to fallback on the attribute_map.