  
      Parameters
      ----------
      plugin: :class:`Model`
          The parent plugin class that owns the IO functions.
      """
      def __init__(self, plugin):
          logger.debug("Initializing %s: (plugin: %s)", self.__class__.__name__, plugin)
          self._model = plugin.model
          self._name = plugin.name
 
</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)
        The parent plugin class that owns the weights functions.
  
      Parameters
      ----------
      plugin: :class:`Model`
        The parent plugin class that owns the weights functions.
      """
      def __init__(self, plugin):
          logger.debug("Initializing %s: (plugin: %s)", self.__class__.__name__, plugin)
          self._model = plugin.model
          self._name = plugin.name
keep keep keep keep replace keep keep keep keep keep
 Bugfix -Weights freezing/loading for dfl-sae 
plugins/train/model/_base.py