          self.add_network("decoder", "b", self.decoder_b(), is_output=True)
          self.add_network("encoder", None, self.encoder())
          logger.debug("Added networks")
  
      def build_autoencoders(self):
          """ Initialize original model """
          logger.debug("Initializing model")
          inputs = [Input(shape=self.input_shape, name="face")]
          if self.config.get("mask_type", None):
              mask_shape = self.config["output_size"], self.config["output_size"], 1
              inputs.append(Input(shape=mask_shape, name="mask"))
 
</s> remove     def build_autoencoders(self):
 </s> add     def build_autoencoders(self, inputs):
 </s> remove         inputs = [Input(shape=self.input_shape, name="face")]
        if self.config.get("mask_type", None):
            mask_shape = self.config["output_size"], self.config["output_size"], 1
            inputs.append(Input(shape=mask_shape, name="mask"))

 </s> add  </s> remove     def build_autoencoders(self):
 </s> add     def build_autoencoders(self, inputs): </s> remove         """ Add the original model weights """
 </s> add         """ Add the realface model weights """ </s> remove         inputs = [Input(shape=self.input_shape, name="face")]
        if self.config.get("mask_type", None):
            mask_shape = (self.input_shape[:2] + (1, ))
            inputs.append(Input(shape=mask_shape, name="mask"))

 </s> add 
    def build_autoencoders(self, inputs):
        """ Initialize realface model """
          self.add_network("decoder", "b", self.decoder_b(), is_output=True)
          self.add_network("encoder", None, self.encoder())
          logger.debug("Added networks")
  
    def build_autoencoders(self, inputs):
        """ Initialize realface model """
    def build_autoencoders(self, inputs):
        """ Initialize realface model """
          logger.debug("Initializing model")
          inputs = [Input(shape=self.input_shape, name="face")]
          if self.config.get("mask_type", None):
              mask_shape = self.config["output_size"], self.config["output_size"], 1
              inputs.append(Input(shape=mask_shape, name="mask"))
keep keep keep keep replace replace keep keep keep keep keep
 Model Structure Changes

Standardize inputs for AutoEncoders
Name outputs from AutoEncoders
Name layers in nn_blocks.py 
plugins/train/model/realface.py