  from keras.layers import Concatenate, Dense, Flatten, Input, LeakyReLU, Reshape
  
  from lib.model.nn_blocks import Conv2DOutput, Conv2DBlock, ResidualBlock, UpscaleBlock
  
  from ._base import ModelBase, KerasModel
  
  
  class Model(ModelBase):
      """ SAE Model from DFL """
      def __init__(self, *args, **kwargs):
 
</s> remove         self.architecture = self.config["architecture"].lower()
 </s> add  </s> remove         The parent plugin class that owns the IO functions.
 </s> add         The parent plugin class that owns the weights functions. </s> remove                                  name="{}_{}".format(self.name, self.architecture))
 </s> add                                  name=self.name)
from ._base import ModelBase, KerasModel, logger
  from keras.layers import Concatenate, Dense, Flatten, Input, LeakyReLU, Reshape
  
  from lib.model.nn_blocks import Conv2DOutput, Conv2DBlock, ResidualBlock, UpscaleBlock
  
from ._base import ModelBase, KerasModel, logger
  
  
  class Model(ModelBase):
      """ SAE Model from DFL """
      def __init__(self, *args, **kwargs):
keep keep keep keep replace keep keep keep keep keep
 Bugfix -Weights freezing/loading for dfl-sae 
plugins/train/model/dfl_sae.py