for TensorRT, Colab T4, tensorrt_diffusion.ipynb