Speed Inference Time for IDM

Hi, I wonder how to speed up inference time of this model. It took ~15s for each time inference. Any solution?