Fix config attribute

leftshiftone · Aug 8, 2023 · fc47cc3 · fc47cc3
1 parent 3fa2531
commit fc47cc3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1 @@
+*.py[cod]
diff --git a/onnx_transformers/model_for_encoding.py b/onnx_transformers/model_for_encoding.py
@@ -47,7 +47,7 @@ def get_output_dim(self) -> int:
 
  if isinstance(shape, str):
  logging.warning("cannot determine output shape from onnx session. try to return dim from config")
- assert "dim" in self.model_config.config, "cannot determine output dim"
- return self.model_config.dim
+ assert "dim" in self.model_config, "cannot determine output dim"
+ return self.model_config.get("dim")
  # noinspection PyTypeChecker
  return shape