测试时间问题

[YLiu-creator]

我根据您提供的代码已经成功运行了自己的数据集，但是在测试阶段每个样本需要消耗大量的时间进行输出预测。请问您是如何解决这个问题的呢？在测试过程中，我看您并没有调整diffusion_steps，也没有用到DDIM或者DPM_solver