diff --git a/modules/onnx_impl/__init__.py b/modules/onnx_impl/__init__.py
index 61ff7c76b..a61e5b50c 100644
--- a/modules/onnx_impl/__init__.py
+++ b/modules/onnx_impl/__init__.py
@@ -105,7 +105,10 @@ class VAEConfig:
         self.config = config
 
     def __getattr__(self, key):
-        return self.config.get(key, VAEConfig.DEFAULTS[key])
+        return self.config.get(key, VAEConfig.DEFAULTS.get(key, None))
+
+    def get(self, key, default):
+        return self.config.get(key, VAEConfig.DEFAULTS.get(key, default))
 
 
 class VAE(TorchCompatibleModule):