          self.input_shape = self._get_input_shape()
          self.color_order = _MODEL_MAPPING[self.config["enc_architecture"]].get("color_order",
                                                                                 "rgb")
  
      def build(self):
          """ Build the model and assign to :attr:`model`.
  
          Override's the default build function for allowing the setting of dropout rate for pre-
          existing models.
          """
 
</s> remove     def _reshape_for_output(self, inputs):
 </s> add     def _reshape_for_output(self, inputs: Tensor) -> Tensor: </s> remove     def __init__(self, input_shape, config):
 </s> add     def __init__(self, input_shape: Tuple[int, int, int], config: dict) -> None: </s> remove     def _build_encoders(self, inputs):
 </s> add     def _build_encoders(self, inputs: List[Tensor]) -> Dict[str, keras.models.Model]: </s> remove     def _build_g_blocks(self, inputs):
 </s> add     def _build_g_blocks(
                self,
                inputs: Dict[str, List[keras.models.Model]]
            ) -> Dict[str, Union[List[keras.models.Model], keras.models.Model]]:
    def build(self) -> None:
          self.input_shape = self._get_input_shape()
          self.color_order = _MODEL_MAPPING[self.config["enc_architecture"]].get("color_order",
                                                                                 "rgb")
  
    def build(self) -> None:
          """ Build the model and assign to :attr:`model`.
  
          Override's the default build function for allowing the setting of dropout rate for pre-
          existing models.
          """
keep keep keep keep replace keep keep keep keep keep
 Model updates
- Increase model summary width
- Phaze A updates
  - Update some min/max values
  - Add Decoder Filter Slope Mode
  - Add additional arguments for Upsampling2D
  - Adjust upsampling method for multiple upsamples in FC layers
  - Typing 
plugins/train/model/phaze_a.py