Merge branch 'master' of https://github.com/DarthAffe/StableDiffusion.NET

Small naming changes in sd.cpp

Header/stable-diffusion.h

@@ -31,46 +31,46 @@ extern "C" {
 enum rng_type_t {
     STD_DEFAULT_RNG,
     CUDA_RNG,
+    CPU_RNG,
     RNG_TYPE_COUNT
 };
 
 enum sample_method_t {
-    SAMPLE_METHOD_DEFAULT,
-    EULER,
-    HEUN,
-    DPM2,
-    DPMPP2S_A,
-    DPMPP2M,
-    DPMPP2Mv2,
-    IPNDM,
-    IPNDM_V,
-    LCM,
-    DDIM_TRAILING,
-    TCD,
-    EULER_A,
+    EULER_SAMPLE_METHOD,
+    EULER_A_SAMPLE_METHOD,
+    HEUN_SAMPLE_METHOD,
+    DPM2_SAMPLE_METHOD,
+    DPMPP2S_A_SAMPLE_METHOD,
+    DPMPP2M_SAMPLE_METHOD,
+    DPMPP2Mv2_SAMPLE_METHOD,
+    IPNDM_SAMPLE_METHOD,
+    IPNDM_V_SAMPLE_METHOD,
+    LCM_SAMPLE_METHOD,
+    DDIM_TRAILING_SAMPLE_METHOD,
+    TCD_SAMPLE_METHOD,
     SAMPLE_METHOD_COUNT
 };
 
 enum scheduler_t {
-    DEFAULT,
-    DISCRETE,
-    KARRAS,
-    EXPONENTIAL,
-    AYS,
-    GITS,
-    SGM_UNIFORM,
-    SIMPLE,
-    SMOOTHSTEP,
-    SCHEDULE_COUNT
+    DISCRETE_SCHEDULER,
+    KARRAS_SCHEDULER,
+    EXPONENTIAL_SCHEDULER,
+    AYS_SCHEDULER,
+    GITS_SCHEDULER,
+    SGM_UNIFORM_SCHEDULER,
+    SIMPLE_SCHEDULER,
+    SMOOTHSTEP_SCHEDULER,
+    LCM_SCHEDULER,
+    SCHEDULER_COUNT
 };
 
 enum prediction_t {
-    DEFAULT_PRED,
     EPS_PRED,
     V_PRED,
     EDM_V_PRED,
-    SD3_FLOW_PRED,
+    FLOW_PRED,
     FLUX_FLOW_PRED,
+    FLUX2_FLOW_PRED,
     PREDICTION_COUNT
 };
 
@@ -156,8 +156,8 @@ typedef struct {
     const char* clip_g_path;
     const char* clip_vision_path;
     const char* t5xxl_path;
-    const char* qwen2vl_path;
-    const char* qwen2vl_vision_path;
+    const char* llm_path;
+    const char* llm_vision_path;
     const char* diffusion_model_path;
     const char* high_noise_diffusion_model_path;
     const char* vae_path;
@@ -166,11 +166,13 @@ typedef struct {
     const char* lora_model_dir;
     const char* embedding_dir;
     const char* photo_maker_path;
+    const char* tensor_type_rules;
     bool vae_decode_only;
     bool free_params_immediately;
     int n_threads;
     enum sd_type_t wtype;
     enum rng_type_t rng_type;
+    enum rng_type_t sampler_rng_type;
     enum prediction_t prediction;
     enum lora_apply_mode_t lora_apply_mode;
     bool offload_params_to_cpu;
@@ -226,6 +228,13 @@ typedef struct {
     float style_strength;
 } sd_pm_params_t;  // photo maker
 
+typedef struct {
+    bool enabled;
+    float reuse_threshold;
+    float start_percent;
+    float end_percent;
+} sd_easycache_params_t;
+
 typedef struct {
     const char* prompt;
     const char* negative_prompt;
@@ -246,6 +255,7 @@ typedef struct {
     float control_strength;
     sd_pm_params_t pm_params;
     sd_tiling_params_t vae_tiling_params;
+    sd_easycache_params_t easycache;
 } sd_img_gen_params_t;
 
 typedef struct {
@@ -265,18 +275,19 @@ typedef struct {
     int64_t seed;
     int video_frames;
     float vace_strength;
+    sd_easycache_params_t easycache;
 } sd_vid_gen_params_t;
 
 typedef struct sd_ctx_t sd_ctx_t;
 
 typedef void (*sd_log_cb_t)(enum sd_log_level_t level, const char* text, void* data);
 typedef void (*sd_progress_cb_t)(int step, int steps, float time, void* data);
-typedef void (*sd_preview_cb_t)(int step, int frame_count, sd_image_t* frames, bool is_noisy);
+typedef void (*sd_preview_cb_t)(int step, int frame_count, sd_image_t* frames, bool is_noisy, void* data);
 
 SD_API void sd_set_log_callback(sd_log_cb_t sd_log_cb, void* data);
 SD_API void sd_set_progress_callback(sd_progress_cb_t cb, void* data);
-SD_API void sd_set_preview_callback(sd_preview_cb_t cb, enum preview_t mode, int interval, bool denoised, bool noisy);
-SD_API int32_t get_num_physical_cores();
+SD_API void sd_set_preview_callback(sd_preview_cb_t cb, enum preview_t mode, int interval, bool denoised, bool noisy, void* data);
+SD_API int32_t sd_get_num_physical_cores();
 SD_API const char* sd_get_system_info();
 
 SD_API const char* sd_type_name(enum sd_type_t type);
@@ -285,8 +296,8 @@ SD_API const char* sd_rng_type_name(enum rng_type_t rng_type);
 SD_API enum rng_type_t str_to_rng_type(const char* str);
 SD_API const char* sd_sample_method_name(enum sample_method_t sample_method);
 SD_API enum sample_method_t str_to_sample_method(const char* str);
-SD_API const char* sd_schedule_name(enum scheduler_t scheduler);
-SD_API enum scheduler_t str_to_schedule(const char* str);
+SD_API const char* sd_scheduler_name(enum scheduler_t scheduler);
+SD_API enum scheduler_t str_to_scheduler(const char* str);
 SD_API const char* sd_prediction_name(enum prediction_t prediction);
 SD_API enum prediction_t str_to_prediction(const char* str);
 SD_API const char* sd_preview_name(enum preview_t preview);
@@ -294,16 +305,20 @@ SD_API enum preview_t str_to_preview(const char* str);
 SD_API const char* sd_lora_apply_mode_name(enum lora_apply_mode_t mode);
 SD_API enum lora_apply_mode_t str_to_lora_apply_mode(const char* str);
 
+SD_API void sd_easycache_params_init(sd_easycache_params_t* easycache_params);
+
 SD_API void sd_ctx_params_init(sd_ctx_params_t* sd_ctx_params);
 SD_API char* sd_ctx_params_to_str(const sd_ctx_params_t* sd_ctx_params);
 
 SD_API sd_ctx_t* new_sd_ctx(const sd_ctx_params_t* sd_ctx_params);
 SD_API void free_sd_ctx(sd_ctx_t* sd_ctx);
-SD_API enum sample_method_t sd_get_default_sample_method(const sd_ctx_t* sd_ctx);
 
 SD_API void sd_sample_params_init(sd_sample_params_t* sample_params);
 SD_API char* sd_sample_params_to_str(const sd_sample_params_t* sample_params);
 
+SD_API enum sample_method_t sd_get_default_sample_method(const sd_ctx_t* sd_ctx);
+SD_API enum scheduler_t sd_get_default_scheduler(const sd_ctx_t* sd_ctx);
+
 SD_API void sd_img_gen_params_init(sd_img_gen_params_t* sd_img_gen_params);
 SD_API char* sd_img_gen_params_to_str(const sd_img_gen_params_t* sd_img_gen_params);
 SD_API sd_image_t* generate_image(sd_ctx_t* sd_ctx, const sd_img_gen_params_t* sd_img_gen_params);
@@ -342,4 +357,4 @@ SD_API bool preprocess_canny(sd_image_t image,
 }
 #endif
 
-#endif  // __STABLE_DIFFUSION_H__
+#endif  // __STABLE_DIFFUSION_H__

StableDiffusion.NET/Enums/Prediction.cs

@@ -2,10 +2,11 @@
 
 public enum Prediction
 {
-    Default,
     EPS,
     V,
     EDM_V,
-    SD3Flow,
-    FluxFlow
+    Flow,
+    FluxFlow,
+    Flux2Flow,
+    Default
 }

StableDiffusion.NET/Enums/RngType.cs

@@ -3,5 +3,6 @@
 public enum RngType
 {
     Standard,
-    Cuda
+    Cuda,
+    Cpu
 }

StableDiffusion.NET/Enums/Sampler.cs

@@ -2,8 +2,8 @@
 
 public enum Sampler
 {
-    Default,
     Euler,
+    Euler_A,
     Heun,
     DPM2,
     DPMPP2SA,
@@ -14,5 +14,5 @@ public enum Sampler
     LCM,
     DDIM_Trailing,
     TCD,
-    Euler_A,
+    Default
 }

StableDiffusion.NET/Enums/Schedule.cs

@@ -2,7 +2,6 @@
 
 public enum Scheduler
 {
-    Default,
     Discrete,
     Karras,
     Exponential,
@@ -10,5 +9,7 @@ public enum Scheduler
     GITS,
     SGM_Uniform,
     Simple,
-    Smoothstep
+    Smoothstep,
+    LCM,
+    Default
 }

StableDiffusion.NET/Models/Parameter/DiffusionModelParameter.cs

@@ -1,4 +1,5 @@
-using JetBrains.Annotations;
+using System;
+using JetBrains.Annotations;
 
 namespace StableDiffusion.NET;
 
@@ -91,6 +92,8 @@ public sealed class DiffusionModelParameter
     /// </summary>
     public RngType RngType { get; set; } = RngType.Standard;
 
+    public RngType SamplerRngType { get; set; } = RngType.Standard;
+
     public Prediction Prediction { get; set; } = Prediction.Default;
 
     public LoraApplyMode LoraApplyMode { get; set; } = LoraApplyMode.Auto;
@@ -110,6 +113,8 @@ public sealed class DiffusionModelParameter
     /// </summary>
     public string StackedIdEmbeddingsDirectory { get; set; } = string.Empty;
 
+    public string TensorTypeRules { get; set; } = string.Empty;
+
     /// <summary>
     /// path to full model
     /// </summary>
@@ -135,9 +140,13 @@ public sealed class DiffusionModelParameter
     /// </summary>
     public string T5xxlPath { get; set; } = string.Empty;
 
-    public string Qwen2VLPath { get; set; } = string.Empty;
+    [Obsolete("Use LLMPath instead")]
+    public string Qwen2VLPath { get => LLMPath; set => LLMPath = value; }
+    public string LLMPath { get; set; } = string.Empty;
 
-    public string Qwen2VLVisionPath { get; set; } = string.Empty;
+    [Obsolete("Use LLMVisionPath instead")]
+    public string Qwen2VLVisionPath { get => LLMVisionPath; set => LLMVisionPath = value; }
+    public string LLMVisionPath { get; set; } = string.Empty;
 
     public string ClipVisionPath { get; set; } = string.Empty;
     public string HighNoiseDiffusionModelPath { get; set; } = string.Empty;

StableDiffusion.NET/Models/Parameter/EasyCache.cs

@@ -0,0 +1,11 @@
+namespace StableDiffusion.NET;
+
+public sealed class EasyCache
+{
+    public bool IsEnabled { get; set; }
+    public float ReuseThreshold { get; set; }
+    public float StartPercent { get; set; }
+    public float EndPercent { get; set; }
+
+    internal EasyCache() { }
+}

StableDiffusion.NET/Models/Parameter/Extensions/DiffusionModelBuilderExtension.cs

@@ -236,16 +236,20 @@ public static class DiffusionModelBuilderExtension
         return parameter;
     }
 
-    public static DiffusionModelParameter WithQwen2VLPath(this DiffusionModelParameter parameter, string qwen2VLPath)
+    [Obsolete("Use WithLLMPath instead")]
+    public static DiffusionModelParameter WithQwen2VLPath(this DiffusionModelParameter parameter, string qwen2VLPath) => parameter.WithLLMPath(qwen2VLPath);
+    public static DiffusionModelParameter WithLLMPath(this DiffusionModelParameter parameter, string llmPath)
     {
-        parameter.Qwen2VLPath = qwen2VLPath;
+        parameter.LLMPath = llmPath;
 
         return parameter;
     }
 
-    public static DiffusionModelParameter WithQwen2VLVisionPath(this DiffusionModelParameter parameter, string qwen2VLVisionPath)
+    [Obsolete("Use WithLLMVisionPath instead")]
+    public static DiffusionModelParameter WithQwen2VLVisionPath(this DiffusionModelParameter parameter, string qwen2VLVisionPath) => parameter.WithLLMVisionPath(qwen2VLVisionPath);
+    public static DiffusionModelParameter WithLLMVisionPath(this DiffusionModelParameter parameter, string llmVisionPath)
     {
-        parameter.Qwen2VLVisionPath = qwen2VLVisionPath;
+        parameter.LLMVisionPath = llmVisionPath;
 
         return parameter;
     }

StableDiffusion.NET/Models/Parameter/ImageGenerationParameter.cs

@@ -59,6 +59,8 @@ public sealed class ImageGenerationParameter
 
     public TilingParameter VaeTiling { get; } = new();
 
+    public EasyCache EasyCache { get; } = new();
+
     #endregion
 
     public static ImageGenerationParameter Create() => new();

StableDiffusion.NET/Models/Parameter/VideoGenerationParameter.cs

@@ -38,6 +38,8 @@ public sealed class VideoGenerationParameter
 
     public float VaceStrength { get; set; }
 
+    public EasyCache EasyCache { get; } = new();
+
     #endregion
 
     public static VideoGenerationParameter Create() => new();

StableDiffusion.NET/Native/Marshaller/DiffusionModelParameterMarshaller.cs

@@ -14,8 +14,8 @@ internal static unsafe class DiffusionModelParameterMarshaller
             clip_g_path = AnsiStringMarshaller.ConvertToUnmanaged(managed.ClipGPath),
             clip_vision_path = AnsiStringMarshaller.ConvertToUnmanaged(managed.ClipVisionPath),
             t5xxl_path = AnsiStringMarshaller.ConvertToUnmanaged(managed.T5xxlPath),
-            qwen2vl_path = AnsiStringMarshaller.ConvertToUnmanaged(managed.Qwen2VLPath),
-            qwen2vl_vision_path = AnsiStringMarshaller.ConvertToUnmanaged(managed.Qwen2VLVisionPath),
+            llm_path = AnsiStringMarshaller.ConvertToUnmanaged(managed.LLMPath),
+            llm_vision_path = AnsiStringMarshaller.ConvertToUnmanaged(managed.LLMVisionPath),
             diffusion_model_path = AnsiStringMarshaller.ConvertToUnmanaged(managed.DiffusionModelPath),
             high_noise_diffusion_model_path = AnsiStringMarshaller.ConvertToUnmanaged(managed.HighNoiseDiffusionModelPath),
             vae_path = AnsiStringMarshaller.ConvertToUnmanaged(managed.VaePath),
@@ -24,11 +24,13 @@ internal static unsafe class DiffusionModelParameterMarshaller
             lora_model_dir = AnsiStringMarshaller.ConvertToUnmanaged(managed.LoraModelDirectory),
             embedding_dir = AnsiStringMarshaller.ConvertToUnmanaged(managed.EmbeddingsDirectory),
             photo_maker_path = AnsiStringMarshaller.ConvertToUnmanaged(managed.StackedIdEmbeddingsDirectory),
+            tensor_type_rules = AnsiStringMarshaller.ConvertToUnmanaged(managed.TensorTypeRules),
             vae_decode_only = (sbyte)(managed.VaeDecodeOnly ? 1 : 0),
             free_params_immediately = (sbyte)(managed.FreeParamsImmediately ? 1 : 0),
             n_threads = managed.ThreadCount,
             wtype = managed.Quantization,
             rng_type = managed.RngType,
+            sampler_rng_type = managed.SamplerRngType,
             prediction = managed.Prediction,
             lora_apply_mode = managed.LoraApplyMode,
             offload_params_to_cpu = (sbyte)(managed.OffloadParamsToCPU ? 1 : 0),
@@ -54,8 +56,8 @@ internal static unsafe class DiffusionModelParameterMarshaller
             ClipGPath = AnsiStringMarshaller.ConvertToManaged(unmanaged.clip_g_path) ?? string.Empty,
             ClipVisionPath = AnsiStringMarshaller.ConvertToManaged(unmanaged.clip_vision_path) ?? string.Empty,
             T5xxlPath = AnsiStringMarshaller.ConvertToManaged(unmanaged.t5xxl_path) ?? string.Empty,
-            Qwen2VLPath = AnsiStringMarshaller.ConvertToManaged(unmanaged.qwen2vl_path) ?? string.Empty,
-            Qwen2VLVisionPath = AnsiStringMarshaller.ConvertToManaged(unmanaged.qwen2vl_vision_path) ?? string.Empty,
+            LLMPath = AnsiStringMarshaller.ConvertToManaged(unmanaged.llm_path) ?? string.Empty,
+            LLMVisionPath = AnsiStringMarshaller.ConvertToManaged(unmanaged.llm_vision_path) ?? string.Empty,
             DiffusionModelPath = AnsiStringMarshaller.ConvertToManaged(unmanaged.diffusion_model_path) ?? string.Empty,
             HighNoiseDiffusionModelPath = AnsiStringMarshaller.ConvertToManaged(unmanaged.high_noise_diffusion_model_path) ?? string.Empty,
             VaePath = AnsiStringMarshaller.ConvertToManaged(unmanaged.vae_path) ?? string.Empty,
@@ -64,11 +66,13 @@ internal static unsafe class DiffusionModelParameterMarshaller
             LoraModelDirectory = AnsiStringMarshaller.ConvertToManaged(unmanaged.lora_model_dir) ?? string.Empty,
             EmbeddingsDirectory = AnsiStringMarshaller.ConvertToManaged(unmanaged.embedding_dir) ?? string.Empty,
             StackedIdEmbeddingsDirectory = AnsiStringMarshaller.ConvertToManaged(unmanaged.photo_maker_path) ?? string.Empty,
+            TensorTypeRules = AnsiStringMarshaller.ConvertToManaged(unmanaged.tensor_type_rules) ?? string.Empty,
             VaeDecodeOnly = unmanaged.vae_decode_only == 1,
             FreeParamsImmediately = unmanaged.free_params_immediately == 1,
             ThreadCount = unmanaged.n_threads,
             Quantization = unmanaged.wtype,
             RngType = unmanaged.rng_type,
+            SamplerRngType = unmanaged.sampler_rng_type,
             Prediction = unmanaged.prediction,
             LoraApplyMode = unmanaged.lora_apply_mode,
             OffloadParamsToCPU = unmanaged.offload_params_to_cpu == 1,
@@ -92,8 +96,8 @@ internal static unsafe class DiffusionModelParameterMarshaller
         AnsiStringMarshaller.Free(unmanaged.clip_l_path);
         AnsiStringMarshaller.Free(unmanaged.clip_g_path);
         AnsiStringMarshaller.Free(unmanaged.t5xxl_path);
-        AnsiStringMarshaller.Free(unmanaged.qwen2vl_path);
-        AnsiStringMarshaller.Free(unmanaged.qwen2vl_vision_path);
+        AnsiStringMarshaller.Free(unmanaged.llm_path);
+        AnsiStringMarshaller.Free(unmanaged.llm_vision_path);
         AnsiStringMarshaller.Free(unmanaged.diffusion_model_path);
         AnsiStringMarshaller.Free(unmanaged.vae_path);
         AnsiStringMarshaller.Free(unmanaged.taesd_path);
@@ -101,5 +105,6 @@ internal static unsafe class DiffusionModelParameterMarshaller
         AnsiStringMarshaller.Free(unmanaged.lora_model_dir);
         AnsiStringMarshaller.Free(unmanaged.embedding_dir);
         AnsiStringMarshaller.Free(unmanaged.photo_maker_path);
+        AnsiStringMarshaller.Free(unmanaged.tensor_type_rules);
     }
 }

StableDiffusion.NET/Native/Marshaller/ImageGenerationParameterMarshaller.cs

@@ -46,6 +46,13 @@ internal static class ImageGenerationParameterMarshaller
                 TargetOverlap = unmanaged.vae_tiling_params.target_overlap,
                 RelSizeX = unmanaged.vae_tiling_params.rel_size_x,
                 RelSizeY = unmanaged.vae_tiling_params.rel_size_y
+            },
+            EasyCache =
+            {
+                IsEnabled = unmanaged.easycache.enabled == 1,
+                ReuseThreshold = unmanaged.easycache.reuse_threshold,
+                StartPercent = unmanaged.easycache.start_percent,
+                EndPercent = unmanaged.easycache.end_percent
             }
         };
 
@@ -126,6 +133,14 @@ internal static class ImageGenerationParameterMarshaller
                 rel_size_y = managed.VaeTiling.RelSizeY
             };
 
+            Native.Types.sd_easycache_params_t easyCache = new()
+            {
+                enabled = (sbyte)(managed.EasyCache.IsEnabled ? 1 : 0),
+                reuse_threshold = managed.EasyCache.ReuseThreshold,
+                start_percent = managed.EasyCache.StartPercent,
+                end_percent = managed.EasyCache.EndPercent,
+            };
+
             _imgGenParams = new Native.Types.sd_img_gen_params_t
             {
                 prompt = AnsiStringMarshaller.ConvertToUnmanaged(managed.Prompt),
@@ -145,7 +160,8 @@ internal static class ImageGenerationParameterMarshaller
                 control_image = _controlNetImage,
                 control_strength = managed.ControlNet.Strength,
                 pm_params = photoMakerParams,
-                vae_tiling_params = tilingParams
+                vae_tiling_params = tilingParams,
+                easycache = easyCache
             };
         }
 

StableDiffusion.NET/Native/Marshaller/VideoGenerationParameterMarshaller.cs

@@ -27,7 +27,14 @@ internal static class VideoGenerationParameterMarshaller
             Strength = unmanaged.strength,
             Seed = unmanaged.seed,
             FrameCount = unmanaged.video_frames,
-            VaceStrength = unmanaged.vace_strength
+            VaceStrength = unmanaged.vace_strength,
+            EasyCache =
+            {
+                IsEnabled = unmanaged.easycache.enabled == 1,
+                ReuseThreshold = unmanaged.easycache.reuse_threshold,
+                StartPercent = unmanaged.easycache.start_percent,
+                EndPercent = unmanaged.easycache.end_percent
+            }
         };
 
         return parameter;
@@ -68,6 +75,14 @@ internal static class VideoGenerationParameterMarshaller
             _initImage = managed.InitImage?.ToSdImage() ?? new Native.Types.sd_image_t();
             _endImage = managed.EndImage?.ToSdImage() ?? new Native.Types.sd_image_t();
             _controlFrames = managed.ControlFrames == null ? null : managed.ControlFrames.ToSdImage();
+            
+            Native.Types.sd_easycache_params_t easyCache = new()
+            {
+                enabled = (sbyte)(managed.EasyCache.IsEnabled ? 1 : 0),
+                reuse_threshold = managed.EasyCache.ReuseThreshold,
+                start_percent = managed.EasyCache.StartPercent,
+                end_percent = managed.EasyCache.EndPercent,
+            };
 
             _vidGenParams = new Native.Types.sd_vid_gen_params_t
             {
@@ -87,6 +102,7 @@ internal static class VideoGenerationParameterMarshaller
                 seed = managed.Seed,
                 video_frames = managed.FrameCount,
                 vace_strength = managed.VaceStrength,
+                easycache = easyCache,
             };
         }
 

StableDiffusion.NET/Native/Native.cs

@@ -25,6 +25,7 @@ using sd_type_t = Quantization;
 using sd_vid_gen_params_t = VideoGenerationParameter;
 using lora_apply_mode_t = LoraApplyMode;
 using preview_t = Preview;
+using sd_easycache_params_t = Native.Types.sd_easycache_params_t;
 using size_t = nuint;
 using uint32_t = uint;
 using uint8_t = byte;
@@ -59,8 +60,8 @@ internal unsafe partial class Native
             public byte* clip_g_path;
             public byte* clip_vision_path;
             public byte* t5xxl_path;
-            public byte* qwen2vl_path;
-            public byte* qwen2vl_vision_path;
+            public byte* llm_path;
+            public byte* llm_vision_path;
             public byte* diffusion_model_path;
             public byte* high_noise_diffusion_model_path;
             public byte* vae_path;
@@ -69,11 +70,13 @@ internal unsafe partial class Native
             public byte* lora_model_dir;
             public byte* embedding_dir;
             public byte* photo_maker_path;
+            public byte* tensor_type_rules;
             public sbyte vae_decode_only;
             public sbyte free_params_immediately;
             public int n_threads;
             public sd_type_t wtype;
             public rng_type_t rng_type;
+            public rng_type_t sampler_rng_type;
             public prediction_t prediction;
             public lora_apply_mode_t lora_apply_mode;
             public sbyte offload_params_to_cpu;
@@ -131,6 +134,7 @@ internal unsafe partial class Native
             public int shifted_timestep;
         }
 
+        [StructLayout(LayoutKind.Sequential)]
         internal struct sd_pm_params_t
         {
             public sd_image_t* id_images;
@@ -139,6 +143,15 @@ internal unsafe partial class Native
             public float style_strength;
         }
 
+        [StructLayout(LayoutKind.Sequential)]
+        internal struct sd_easycache_params_t
+        {
+            public sbyte enabled;
+            public float reuse_threshold;
+            public float start_percent;
+            public float end_percent;
+        }
+
         [StructLayout(LayoutKind.Sequential)]
         internal struct sd_img_gen_params_t
         {
@@ -161,6 +174,7 @@ internal unsafe partial class Native
             public float control_strength;
             public sd_pm_params_t pm_params;
             public sd_tiling_params_t vae_tiling_params;
+            public sd_easycache_params_t easycache;
         }
 
         [StructLayout(LayoutKind.Sequential)]
@@ -182,6 +196,7 @@ internal unsafe partial class Native
             public int64_t seed;
             public int video_frames;
             public float vace_strength;
+            public sd_easycache_params_t easycache;
         }
 
         internal struct sd_ctx_t;
@@ -192,7 +207,7 @@ internal unsafe partial class Native
 
     internal delegate void sd_log_cb_t(sd_log_level_t level, [MarshalAs(UnmanagedType.LPStr)] string text, void* data);
     internal delegate void sd_progress_cb_t(int step, int steps, float time, void* data);
-    internal delegate void sd_preview_cb_t(int step, int frame_count, sd_image_t* frames, bool is_noisy);
+    internal delegate void sd_preview_cb_t(int step, int frame_count, sd_image_t* frames, bool is_noisy, void* data);
 
     #endregion
 
@@ -205,10 +220,10 @@ internal unsafe partial class Native
     internal static partial void sd_set_progress_callback(sd_progress_cb_t cb, void* data);
 
     [LibraryImport(LIB_NAME, EntryPoint = "sd_set_preview_callback")]
-    internal static partial void sd_set_preview_callback(sd_preview_cb_t? cb, preview_t mode, int interval, [MarshalAs(UnmanagedType.I1)] bool denoised, [MarshalAs(UnmanagedType.I1)] bool noisy);
+    internal static partial void sd_set_preview_callback(sd_preview_cb_t? cb, preview_t mode, int interval, [MarshalAs(UnmanagedType.I1)] bool denoised, [MarshalAs(UnmanagedType.I1)] bool noisy, void* data);
 
-    [LibraryImport(LIB_NAME, EntryPoint = "get_num_physical_cores")]
-    internal static partial int32_t get_num_physical_cores();
+    [LibraryImport(LIB_NAME, EntryPoint = "sd_get_num_physical_cores")]
+    internal static partial int32_t sd_get_num_physical_cores();
 
     [LibraryImport(LIB_NAME, EntryPoint = "sd_get_system_info")]
     [return: MarshalAs(UnmanagedType.LPStr)]
@@ -237,12 +252,12 @@ internal unsafe partial class Native
     [LibraryImport(LIB_NAME, EntryPoint = "str_to_sample_method")]
     internal static partial sample_method_t str_to_sample_method([MarshalAs(UnmanagedType.LPStr)] string str);
 
-    [LibraryImport(LIB_NAME, EntryPoint = "sd_schedule_name")]
+    [LibraryImport(LIB_NAME, EntryPoint = "sd_scheduler_name")]
     [return: MarshalAs(UnmanagedType.LPStr)]
-    internal static partial string sd_schedule_name(scheduler_t schedule);
+    internal static partial string sd_scheduler_name(scheduler_t schedule);
 
-    [LibraryImport(LIB_NAME, EntryPoint = "str_to_schedule")]
-    internal static partial scheduler_t str_to_schedule([MarshalAs(UnmanagedType.LPStr)] string str);
+    [LibraryImport(LIB_NAME, EntryPoint = "str_to_scheduler")]
+    internal static partial scheduler_t str_to_scheduler([MarshalAs(UnmanagedType.LPStr)] string str);
 
     [LibraryImport(LIB_NAME, EntryPoint = "sd_prediction_name")]
     [return: MarshalAs(UnmanagedType.LPStr)]
@@ -265,6 +280,9 @@ internal unsafe partial class Native
     [LibraryImport(LIB_NAME, EntryPoint = "str_to_lora_apply_mode")]
     internal static partial lora_apply_mode_t str_to_lora_apply_mode([MarshalAs(UnmanagedType.LPStr)] string str);
 
+    [LibraryImport(LIB_NAME, EntryPoint = "sd_easycache_params_init")]
+    internal static partial void sd_easycache_params_init(ref sd_easycache_params_t easycache_params);
+
     //
 
     [LibraryImport(LIB_NAME, EntryPoint = "sd_ctx_params_init")]
@@ -282,9 +300,6 @@ internal unsafe partial class Native
     [LibraryImport(LIB_NAME, EntryPoint = "free_sd_ctx")]
     internal static partial void free_sd_ctx(sd_ctx_t* sd_ctx);
 
-    [LibraryImport(LIB_NAME, EntryPoint = "sd_get_default_sample_method")]
-    internal static partial sample_method_t sd_get_default_sample_method(sd_ctx_t* sd_ctx);
-
     //
 
     [LibraryImport(LIB_NAME, EntryPoint = "sd_sample_params_init")]
@@ -301,6 +316,12 @@ internal unsafe partial class Native
     [return: MarshalAs(UnmanagedType.LPStr)]
     internal static partial string sd_img_gen_params_to_str([MarshalUsing(typeof(ImageGenerationParameterMarshaller))] in sd_img_gen_params_t sd_img_gen_params);
 
+    [LibraryImport(LIB_NAME, EntryPoint = "sd_get_default_sample_method")]
+    internal static partial sample_method_t sd_get_default_sample_method(sd_ctx_t* sd_ctx);
+
+    [LibraryImport(LIB_NAME, EntryPoint = "sd_get_default_scheduler")]
+    internal static partial scheduler_t sd_get_default_scheduler(sd_ctx_t* sd_ctx);
+
     [LibraryImport(LIB_NAME, EntryPoint = "generate_image")]
     internal static partial sd_image_t* generate_image(sd_ctx_t* sd_ctx, [MarshalUsing(typeof(ImageGenerationParameterMarshaller))] in sd_img_gen_params_t sd_img_gen_params);
 

StableDiffusion.NET/StableDiffusionCpp.cs

@@ -45,7 +45,7 @@ public static unsafe class StableDiffusionCpp
         else if (_previewCallback == null)
             _previewCallback = OnPreview;
 
-        Native.sd_set_preview_callback(_previewCallback, mode, interval, denoised, noisy);
+        Native.sd_set_preview_callback(_previewCallback, mode, interval, denoised, noisy, null);
     }
 
     public static void Convert(string modelPath, string vaePath, Quantization quantization, string outputPath, string tensorTypeRules = "")
@@ -60,7 +60,7 @@ public static unsafe class StableDiffusionCpp
 
     public static string GetSystemInfo() => Native.sd_get_system_info();
 
-    public static int GetNumPhysicalCores() => Native.get_num_physical_cores();
+    public static int GetNumPhysicalCores() => Native.sd_get_num_physical_cores();
 
     public static Image<ColorRGB> PreprocessCanny(CannyParameter parameter)
     {
@@ -104,7 +104,7 @@ public static unsafe class StableDiffusionCpp
         catch { /**/ }
     }
 
-    private static void OnPreview(int step, int frameCount, Native.Types.sd_image_t* frames, bool isNoisy)
+    private static void OnPreview(int step, int frameCount, Native.Types.sd_image_t* frames, bool isNoisy, void* data)
     {
         try
         {