          logger.info("Saving Tensorflow 2.x model to '%s'", filename)
          new_model.save(filename)
          # Penalized Loss and Learn Mask used to be disabled automatically if a mask wasn't
          # selected, so disable it if enabled, but mask_type is None
          if self.config["penalized_mask_loss"] and self.config["mask_type"] is None:
              self.config["penalized_mask_loss"] = False
          if self.config["learn_mask"] and self.config["mask_type"] is None:
              self.config["learn_mask"] = False
          self._state.save()
  
 
</s> remove         if self.config["learn_mask"] and self.config["mask_type"] is None:
 </s> add  </s> remove             logger.info("Updating legacy weights from '%s'...", model_name)
 </s> add 
        if self.config["mask_type"] is None:
          logger.info("Saving Tensorflow 2.x model to '%s'", filename)
          new_model.save(filename)
          # Penalized Loss and Learn Mask used to be disabled automatically if a mask wasn't
          # selected, so disable it if enabled, but mask_type is None
        if self.config["mask_type"] is None:
              self.config["penalized_mask_loss"] = False
          if self.config["learn_mask"] and self.config["mask_type"] is None:
              self.config["learn_mask"] = False
          self._state.save()
  
keep keep keep keep replace keep keep keep keep keep
 Bugfixes:
  - dfl_h128 Legacy Weights update
  - lib.faces_detect - Logging fix 
plugins/train/model/_base.py