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
    )