@@ -48,12 +48,10 @@ enum sample_method_t {
4848 LCM_SAMPLE_METHOD ,
4949 DDIM_TRAILING_SAMPLE_METHOD ,
5050 TCD_SAMPLE_METHOD ,
51-
5251 SAMPLE_METHOD_COUNT
5352};
5453
5554enum scheduler_t {
56-
5755 DISCRETE_SCHEDULER ,
5856 KARRAS_SCHEDULER ,
5957 EXPONENTIAL_SCHEDULER ,
@@ -73,6 +71,7 @@ enum prediction_t {
7371 EDM_V_PRED ,
7472 SD3_FLOW_PRED ,
7573 FLUX_FLOW_PRED ,
74+ FLUX2_FLOW_PRED ,
7675 PREDICTION_COUNT
7776};
7877
@@ -158,8 +157,8 @@ typedef struct {
158157 const char * clip_g_path ;
159158 const char * clip_vision_path ;
160159 const char * t5xxl_path ;
161- const char * qwen2vl_path ;
162- const char * qwen2vl_vision_path ;
160+ const char * llm_path ;
161+ const char * llm_vision_path ;
163162 const char * diffusion_model_path ;
164163 const char * high_noise_diffusion_model_path ;
165164 const char * vae_path ;
@@ -284,11 +283,11 @@ typedef struct sd_ctx_t sd_ctx_t;
284283
285284typedef void (* sd_log_cb_t )(enum sd_log_level_t level , const char * text , void * data );
286285typedef void (* sd_progress_cb_t )(int step , int steps , float time , void * data );
287- typedef void (* sd_preview_cb_t )(int step , int frame_count , sd_image_t * frames , bool is_noisy );
286+ typedef void (* sd_preview_cb_t )(int step , int frame_count , sd_image_t * frames , bool is_noisy , void * data );
288287
289288SD_API void sd_set_log_callback (sd_log_cb_t sd_log_cb , void * data );
290289SD_API void sd_set_progress_callback (sd_progress_cb_t cb , void * data );
291- SD_API void sd_set_preview_callback (sd_preview_cb_t cb , enum preview_t mode , int interval , bool denoised , bool noisy );
290+ SD_API void sd_set_preview_callback (sd_preview_cb_t cb , enum preview_t mode , int interval , bool denoised , bool noisy , void * data );
292291SD_API int32_t get_num_physical_cores ();
293292SD_API const char * sd_get_system_info ();
294293
@@ -314,7 +313,6 @@ SD_API char* sd_ctx_params_to_str(const sd_ctx_params_t* sd_ctx_params);
314313
315314SD_API sd_ctx_t * new_sd_ctx (const sd_ctx_params_t * sd_ctx_params );
316315SD_API void free_sd_ctx (sd_ctx_t * sd_ctx );
317-
318316
319317SD_API void sd_sample_params_init (sd_sample_params_t * sample_params );
320318SD_API char * sd_sample_params_to_str (const sd_sample_params_t * sample_params );
@@ -360,4 +358,4 @@ SD_API bool preprocess_canny(sd_image_t image,
360358}
361359#endif
362360
363- #endif // __STABLE_DIFFUSION_H__
361+ #endif // __STABLE_DIFFUSION_H__
0 commit comments