diff --git a/lite/trt/sd/trt_unet.cpp b/lite/trt/sd/trt_unet.cpp index 9f2722e5..e48a2544 100644 --- a/lite/trt/sd/trt_unet.cpp +++ b/lite/trt/sd/trt_unet.cpp @@ -185,9 +185,6 @@ void TRTUNet::inference(const std::vector> &clip_output, std: scheduler.step(noise_pred,noise_pred_dims, latents_fp32, noise_pred_dims, pred_sample, t); - scheduler.step(noise_pred,noise_pred_dims, latents_fp32, noise_pred_dims, - pred_sample, t); - std::vector pred_sample_fp16(pred_sample.size(),0); std::transform(pred_sample.begin(), pred_sample.end(), pred_sample_fp16.begin(),[](float f) { return __float2half(f);});