Predicting runoff using rainfall time series data

The dataset contains 2 columns, rainfall and runoff. I have implemented LSTM, GRU and Transformer model to predict the runoff based on rainfall data. I need someone to implement the time series transformer model using TSAI( Time series AI ). TSAI is a API ML library. I can provide more documentation for this.