Skip to content

Commit 66568ff

Browse files
committed
Updated ref-header
1 parent 1dcbe72 commit 66568ff

1 file changed

Lines changed: 6 additions & 8 deletions

File tree

Header/stable-diffusion.h

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -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

5554
enum 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

285284
typedef void (*sd_log_cb_t)(enum sd_log_level_t level, const char* text, void* data);
286285
typedef 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

289288
SD_API void sd_set_log_callback(sd_log_cb_t sd_log_cb, void* data);
290289
SD_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);
292291
SD_API int32_t get_num_physical_cores();
293292
SD_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

315314
SD_API sd_ctx_t* new_sd_ctx(const sd_ctx_params_t* sd_ctx_params);
316315
SD_API void free_sd_ctx(sd_ctx_t* sd_ctx);
317-
318316

319317
SD_API void sd_sample_params_init(sd_sample_params_t* sample_params);
320318
SD_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

Comments
 (0)