Skip to content

Commit 7d3e8c3

Browse files
committed
updated reference header
1 parent 4b5d4c9 commit 7d3e8c3

1 file changed

Lines changed: 23 additions & 0 deletions

File tree

Header/stable-diffusion.h

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -126,6 +126,21 @@ enum sd_log_level_t {
126126
SD_LOG_ERROR
127127
};
128128

129+
enum preview_t {
130+
PREVIEW_NONE,
131+
PREVIEW_PROJ,
132+
PREVIEW_TAE,
133+
PREVIEW_VAE,
134+
PREVIEW_COUNT
135+
};
136+
137+
enum lora_apply_mode_t {
138+
LORA_APPLY_AUTO,
139+
LORA_APPLY_IMMEDIATELY,
140+
LORA_APPLY_AT_RUNTIME,
141+
LORA_APPLY_MODE_COUNT,
142+
};
143+
129144
typedef struct {
130145
bool enabled;
131146
int tile_size_x;
@@ -157,11 +172,13 @@ typedef struct {
157172
enum sd_type_t wtype;
158173
enum rng_type_t rng_type;
159174
enum prediction_t prediction;
175+
enum lora_apply_mode_t lora_apply_mode;
160176
bool offload_params_to_cpu;
161177
bool keep_clip_on_cpu;
162178
bool keep_control_net_on_cpu;
163179
bool keep_vae_on_cpu;
164180
bool diffusion_flash_attn;
181+
bool tae_preview_only;
165182
bool diffusion_conv_direct;
166183
bool vae_conv_direct;
167184
bool force_sdxl_vae_conv_scale;
@@ -254,9 +271,11 @@ typedef struct sd_ctx_t sd_ctx_t;
254271

255272
typedef void (*sd_log_cb_t)(enum sd_log_level_t level, const char* text, void* data);
256273
typedef void (*sd_progress_cb_t)(int step, int steps, float time, void* data);
274+
typedef void (*sd_preview_cb_t)(int step, int frame_count, sd_image_t* frames, bool is_noisy);
257275

258276
SD_API void sd_set_log_callback(sd_log_cb_t sd_log_cb, void* data);
259277
SD_API void sd_set_progress_callback(sd_progress_cb_t cb, void* data);
278+
SD_API void sd_set_preview_callback(sd_preview_cb_t cb, enum preview_t mode, int interval, bool denoised, bool noisy);
260279
SD_API int32_t get_num_physical_cores();
261280
SD_API const char* sd_get_system_info();
262281

@@ -270,6 +289,10 @@ SD_API const char* sd_schedule_name(enum scheduler_t scheduler);
270289
SD_API enum scheduler_t str_to_schedule(const char* str);
271290
SD_API const char* sd_prediction_name(enum prediction_t prediction);
272291
SD_API enum prediction_t str_to_prediction(const char* str);
292+
SD_API const char* sd_preview_name(enum preview_t preview);
293+
SD_API enum preview_t str_to_preview(const char* str);
294+
SD_API const char* sd_lora_apply_mode_name(enum lora_apply_mode_t mode);
295+
SD_API enum lora_apply_mode_t str_to_lora_apply_mode(const char* str);
273296

274297
SD_API void sd_ctx_params_init(sd_ctx_params_t* sd_ctx_params);
275298
SD_API char* sd_ctx_params_to_str(const sd_ctx_params_t* sd_ctx_params);

0 commit comments

Comments
 (0)