          end_idx = None if end_idx == -1 else end_idx
  
          if start_idx == 0:
              var_x = self._reshape_for_output(var_x)
  
              if self._config["learn_mask"]:
                  var_y = self._reshape_for_output(var_y)
  
 
</s> remove         inputs = Input(shape=self._input_shape) if inputs is None else inputs
        var_x = inputs
 </s> add  </s> remove                 var_y = inputs
 </s> add  </s> remove         :class:`keras.models.Model`
            The Decoder model
 </s> add          Tensor or list of tensors
            The output of encoder blocks. Either a single tensor (if learn mask is not enabled) or
            list of tensors (if learn mask is enabled) </s> remove                                  self._input_shape,
 </s> add  </s> remove                                  layer_indicies=indicies)(var_x)
 </s> add                                  layer_indicies=indicies)(inputs)
        if self._config["learn_mask"] and start_idx == 0:
            # Mask needs to be created
            var_x = inputs
            var_y = inputs
        elif self._config["learn_mask"]:
            # Mask has already been created and is an input to upscale blocks
            var_x, var_y = inputs
        else:
            # No mask required
            var_x = inputs

          end_idx = None if end_idx == -1 else end_idx
  
        if self._config["learn_mask"] and start_idx == 0:
            # Mask needs to be created
            var_x = inputs
            var_y = inputs
        elif self._config["learn_mask"]:
            # Mask has already been created and is an input to upscale blocks
            var_x, var_y = inputs
        else:
            # No mask required
            var_x = inputs

          if start_idx == 0:
              var_x = self._reshape_for_output(var_x)
  
              if self._config["learn_mask"]:
                  var_y = self._reshape_for_output(var_y)
  
keep add keep keep keep keep keep keep
 Bugfix: Phaze-A Fix learn mask for upscales in FC 
plugins/train/model/phaze_a.py