How to training DTLN-aec model

[undermore]

Hi, breizhn. thanks for this repository!
I am a beginner in machine learning.
following the readme I have trained DTLN model with 48K audio data and it sames work correctly by running real time processing test with TFlite. I can also run AEC with pretrained DTLN-aec model on 16K audio.

but when I running AEC with DTLN model (whether 16k or 48k) the error shown as below:

/run_aec.py", line 116, in process_file
interpreter_1. set_tensor(input_details_1[2]["index"], lpb_mag)
IndexError: list index out of range

it's caused by the different model params I guess.
I'm not sure if I was missing something.

my params is:

self. fs = 48000
self. batchsize = 22
self. len_samples = 15
self. activation = 'sigmoid'
self. numUnits = 128
self. numLayer = 2
self. blockLen = 512 * 3
self. block_shift = 128 * 3
self. dropout = 0.25
self. lr = 1e-3
self. max_epochs = 3 #just for test
self. encoder_size = 256
self. eps = 1e-7

The question is how can I get a DTLN model for AEC？
Thanks a lot!