b6899e8fc2
* Add Euler sampler * Add Heun sampler * Add DPM++ (2M) sampler * Add modified DPM++ (2M) "v2" sampler. This was proposed in a issue discussion of the stable diffusion webui, at https://github.com/AUTOMATIC1111/stable-diffusion-webui/discussions/8457 and apparently works around overstepping of the DPM++ (2M) method with small step counts. The parameter is called dpmpp2mv2 here. * match code style --------- Co-authored-by: Urs Ganse <urs@nerd2nerd.org> Co-authored-by: leejet <leejet714@gmail.com>
65 lines
1.4 KiB
C++
65 lines
1.4 KiB
C++
#ifndef __STABLE_DIFFUSION_H__
|
|
#define __STABLE_DIFFUSION_H__
|
|
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
enum SDLogLevel {
|
|
DEBUG,
|
|
INFO,
|
|
WARN,
|
|
ERROR
|
|
};
|
|
|
|
enum RNGType {
|
|
STD_DEFAULT_RNG,
|
|
CUDA_RNG
|
|
};
|
|
|
|
enum SampleMethod {
|
|
EULER_A,
|
|
EULER,
|
|
HEUN,
|
|
DPMPP2M,
|
|
DPMPP2Mv2,
|
|
N_SAMPLE_METHODS
|
|
};
|
|
|
|
class StableDiffusionGGML;
|
|
|
|
class StableDiffusion {
|
|
private:
|
|
std::shared_ptr<StableDiffusionGGML> sd;
|
|
|
|
public:
|
|
StableDiffusion(int n_threads = -1,
|
|
bool vae_decode_only = false,
|
|
bool free_params_immediately = false,
|
|
RNGType rng_type = STD_DEFAULT_RNG);
|
|
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,
|
|
int64_t 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,
|
|
int64_t seed);
|
|
};
|
|
|
|
void set_sd_log_level(SDLogLevel level);
|
|
std::string sd_get_system_info();
|
|
|
|
#endif // __STABLE_DIFFUSION_H__
|