Files
stable-diffusion.cpp/src/generation_info.h
T

24 lines
929 B
C++

#ifndef __SD_GENERATION_INFO_H__
#define __SD_GENERATION_INFO_H__
#include <vector>
#include "json.hpp"
#include "stable-diffusion.h"
namespace sd::pipeline {
struct GenerationRequest;
}
namespace sd {
nlohmann::json generation_info(const pipeline::GenerationRequest& request, const sd_img_gen_params_t& params);
nlohmann::json generation_info(const pipeline::GenerationRequest& request, const sd_vid_gen_params_t& params);
nlohmann::json adetailer_generation_info(const sd_adetailer_params_t& params, const sd_img_gen_params_t& inpaint_params);
nlohmann::json upscale_generation_info(uint32_t upscale_factor);
char* copy_generation_info(const nlohmann::json& info);
// Consumes images even on failure.
bool make_image_generation_result(sd_image_t* images, int count, const std::vector<nlohmann::json>& infos, sd_image_generation_result_t** result_out);
}
#endif // __SD_GENERATION_INFO_H__