apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-19 23:16:15 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Add trt backend

Browse Source
This commit is contained in:
zhoushunjie
2022-12-05 07:48:51 +00:00
parent 65f270df29
commit a88c6a6905
3 changed files with 95 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/multimodal/stable_diffusion/cpp/pipeline_stable_diffusion_inpaint.cc
View File

@@ -314,9 +314,6 @@ void StableDiffusionInpaintPipeline::Predict(
vision::FDMat mask_fdmat_t = vision::FDMat::Create((*output_images)[i]);
vision::RGB2BGR::Run(&mask_fdmat_t, vision::ProcLib::OPENCV);
mask_fdmat_t.CopyToTensor(&(*output_images)[i]);
FDTensor sum;
function::Sum((*output_images)[i], &sum, {}, false, true);
FDINFO << "sum = " << ((float*)sum.Data())[0] << std::endl;
}
}
} // namespace fastdeploy