          self.decoder_dim = self.config["decoder_dims"]
  
      @property
      def ae_dims(self):
          """ Set the Autoencoder Dimensions or set to default """
          retval = self.config["autoencoder_dims"]
 
</s> remove         self.architecture = self.config["architecture"].lower()
 </s> add  </s> remove         mappings = dict(df={"{}_encoder.h5".format(self.name): "encoder_df",
                            "{}_decoder_A.h5".format(self.name): "decoder_a",
                            "{}_decoder_B.h5".format(self.name): "decoder_b"},
                        liae={"{}_encoder.h5".format(self.name): "encoder_liae",
                              "{}_intermediate_B.h5".format(self.name): "intermediate_both",
                              "{}_intermediate.h5".format(self.name): "intermediate_b",
                              "{}_decoder.h5".format(self.name): "decoder_both"})
 </s> add         name = "dfl_sae"
        mappings = dict(df={"{}_encoder.h5".format(name): "encoder_df",
                            "{}_decoder_A.h5".format(name): "decoder_a",
                            "{}_decoder_B.h5".format(name): "decoder_b"},
                        liae={"{}_encoder.h5".format(name): "encoder_liae",
                              "{}_intermediate_B.h5".format(name): "intermediate_both",
                              "{}_intermediate.h5".format(name): "intermediate_b",
                              "{}_decoder.h5".format(name): "decoder_both"}) </s> remove                                  name="{}_{}".format(self.name, self.architecture))
 </s> add                                  name=self.name) </s> remove         The parent plugin class that owns the IO functions.
 </s> add         The parent plugin class that owns the weights functions.
    @property
    def name(self):
        """ str: The name of this model based on the plugin name. Overridden as DFL-SAE is
        named differently depending on the architecture selected. """
        basename = super().name
        name = f"{basename}_{self.architecture}"
        return name

          self.decoder_dim = self.config["decoder_dims"]
  
    @property
    def name(self):
        """ str: The name of this model based on the plugin name. Overridden as DFL-SAE is
        named differently depending on the architecture selected. """
        basename = super().name
        name = f"{basename}_{self.architecture}"
        return name

      @property
      def ae_dims(self):
          """ Set the Autoencoder Dimensions or set to default """
          retval = self.config["autoencoder_dims"]
keep add keep keep keep keep
 Bugfix -Weights freezing/loading for dfl-sae 
plugins/train/model/dfl_sae.py