def METHOD_NAME(self, incremental_state: Optional[Dict[str, Dict[str, Optional[Tensor]]]], new_buffer):
    return utils.set_incremental_state(
        self, incremental_state, "input_buffer", new_buffer
    )