@@ -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+
129144typedef 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
255272typedef void (* sd_log_cb_t )(enum sd_log_level_t level , const char * text , void * data );
256273typedef 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
258276SD_API void sd_set_log_callback (sd_log_cb_t sd_log_cb , void * data );
259277SD_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 );
260279SD_API int32_t get_num_physical_cores ();
261280SD_API const char * sd_get_system_info ();
262281
@@ -270,6 +289,10 @@ SD_API const char* sd_schedule_name(enum scheduler_t scheduler);
270289SD_API enum scheduler_t str_to_schedule (const char * str );
271290SD_API const char * sd_prediction_name (enum prediction_t prediction );
272291SD_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
274297SD_API void sd_ctx_params_init (sd_ctx_params_t * sd_ctx_params );
275298SD_API char * sd_ctx_params_to_str (const sd_ctx_params_t * sd_ctx_params );
0 commit comments