Fix model wrapper attribute access passthrough

2024-10-27 12:23:41 -06:00
parent 8dc8d44714
commit c39974786f
1 changed files with 1 additions and 1 deletions
@@ -14,7 +14,7 @@ class JankDiffuseHighModelWrapper:

    def __getattr__(self, key):
        try:
-            return getattr(self.__guided_model, key)
+            return getattr(self.__guided_model.dhso.model, key)
        except AttributeError:
            raise AttributeError(key) from None