Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix onnx vae

Browse Source
This commit is contained in:
Seunghoon Lee
2024-04-06 21:58:54 +09:00
parent b5fa476f08
commit a44cedec0a
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/onnx_impl/__init__.py
+4 -1
View File
@@ -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):