          if self.config["mask_type"] is None:
              self.config["penalized_mask_loss"] = False
              self.config["learn_mask"] = False
          self._state.save()
  
      def _validate_input_shape(self):
          """ Validate that the input shape is either a single shape tuple of 3 dimensions or
          a list of 2 shape tuples of 3 dimensions. """
 
</s> remove         if self.config["learn_mask"] and self.config["mask_type"] is None:
 </s> add  </s> remove         if self.config["penalized_mask_loss"] and self.config["mask_type"] is None:
 </s> add         if self.config["mask_type"] is None: </s> remove             logger.info("Updating legacy weights from '%s'...", model_name)
 </s> add  </s> remove                      size, points, dilation, blur_kernel)
 </s> add                      self.__class__.__name__, size, points, dilation, blur_kernel)
            self.config["eye_multiplier"] = 1
            self.config["mouth_multiplier"] = 1
          if self.config["mask_type"] is None:
              self.config["penalized_mask_loss"] = False
              self.config["learn_mask"] = False
            self.config["eye_multiplier"] = 1
            self.config["mouth_multiplier"] = 1
          self._state.save()
  
      def _validate_input_shape(self):
          """ Validate that the input shape is either a single shape tuple of 3 dimensions or
          a list of 2 shape tuples of 3 dimensions. """
keep keep add keep keep keep keep keep
 Bugfixes:
  - dfl_h128 Legacy Weights update
  - lib.faces_detect - Logging fix 
plugins/train/model/_base.py