Fix error

hydra_node/models.py

@@ -274,7 +274,7 @@ class StableDiffusionModel(nn.Module):
             'masks': None,
             'output': None,
         }
-        return ConfigClass(dict_config)
+        return DotMap(dict_config)
 
     def fuse_model(self, requires_grad=False):
         ema = self.ema