53 lines
1.1 KiB
C++
53 lines
1.1 KiB
C++
#ifndef __STABLE_DIFFUSION_H__
|
|
#define __STABLE_DIFFUSION_H__
|
|
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
enum class SDLogLevel {
|
|
DEBUG,
|
|
INFO,
|
|
WARN,
|
|
ERROR
|
|
};
|
|
|
|
enum SampleMethod {
|
|
EULAR_A,
|
|
};
|
|
|
|
class StableDiffusionGGML;
|
|
|
|
class StableDiffusion {
|
|
private:
|
|
std::shared_ptr<StableDiffusionGGML> sd;
|
|
|
|
public:
|
|
StableDiffusion(int n_threads = -1,
|
|
bool vae_decode_only = false);
|
|
bool load_from_file(const std::string& file_path);
|
|
std::vector<uint8_t> txt2img(
|
|
const std::string& prompt,
|
|
const std::string& negative_prompt,
|
|
float cfg_scale,
|
|
int width,
|
|
int height,
|
|
SampleMethod sample_method,
|
|
int sample_steps,
|
|
int seed);
|
|
std::vector<uint8_t> img2img(
|
|
const std::vector<uint8_t>& init_img,
|
|
const std::string& prompt,
|
|
const std::string& negative_prompt,
|
|
float cfg_scale,
|
|
int width,
|
|
int height,
|
|
SampleMethod sample_method,
|
|
int sample_steps,
|
|
float strength,
|
|
int seed);
|
|
};
|
|
|
|
void set_sd_log_level(SDLogLevel level);
|
|
std::string sd_get_system_info();
|
|
|
|
#endif // __STABLE_DIFFUSION_H__
|