apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-07 17:41:52 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Add example for initial noise input

Browse Source
This commit is contained in:
zhoushunjie
2022-12-05 10:19:44 +00:00
parent 49559a8484
commit 1080d350b1
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
examples/multimodal/stable_diffusion/cpp/main.cc
View File

@@ -205,6 +205,14 @@ int main() {
auto mask_image = cv::imread("overture-creations-mask.png"); auto mask_image = cv::imread("overture-creations-mask.png");
// 8. Predict // 8. Predict
/*
* One may need to pass the initial noise to predict api.
* There's an example:
* std::vector<float> latents_data = {xxxx};
* fastdeploy::FDTensor latents;
* latents.SetExternalData({batch_size * num_images_per_prompt, latents_channels, height / 8, width / 8},fastdeploy::FDDataType::FP32, latents_data.data());
* pipe.Predict(..., /* latents = *\/ &latents, ....);
*/
std::vector<std::string> prompts = { std::vector<std::string> prompts = {
"Face of a yellow cat, high resolution, sitting on a park bench"}; "Face of a yellow cat, high resolution, sitting on a park bench"};
std::vector<fastdeploy::FDTensor> outputs; std::vector<fastdeploy::FDTensor> outputs;