                   side: Literal["a", "b", "both", "shared"],
                   input_shape: Tuple[int, int, int],
                   config: dict,
                   layer_indicies: Optional[Tuple[int, int]] = None) -> None:
          logger.debug("Initializing: %s (side: %s, input_shape: %s, layer_indicies: %s)",
                       self.__class__.__name__, side, input_shape, layer_indicies)
          self._side = side
          self._input_shape = input_shape
          self._config = config
          self._is_dny = self._config["dec_upscale_method"].lower() == "upscale_dny"
          self._layer_indicies = layer_indicies
 
</s> remove                  input_shape: Tuple[int, int, int],
 </s> add  </s> remove         self._input_shape = input_shape
 </s> add  </s> remove     def __init__(self, input_shape: Tuple[int, int, int], config: dict) -> None:
 </s> add     def __init__(self, input_shape: Tuple[int, ...], config: dict) -> None: </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 
        logger.debug("Initializing: %s (side: %s, layer_indicies: %s)",
                     self.__class__.__name__, side, layer_indicies)
                   side: Literal["a", "b", "both", "shared"],
                   input_shape: Tuple[int, int, int],
                   config: dict,
                   layer_indicies: Optional[Tuple[int, int]] = None) -> None:
        logger.debug("Initializing: %s (side: %s, layer_indicies: %s)",
                     self.__class__.__name__, side, layer_indicies)
        logger.debug("Initializing: %s (side: %s, layer_indicies: %s)",
                     self.__class__.__name__, side, layer_indicies)
          self._side = side
          self._input_shape = input_shape
          self._config = config
          self._is_dny = self._config["dec_upscale_method"].lower() == "upscale_dny"
          self._layer_indicies = layer_indicies
keep keep keep keep replace replace keep keep keep keep keep
 Bugfix: Phaze-A Fix learn mask for upscales in FC 
plugins/train/model/phaze_a.py