diff --git a/StableDiffusion.NET/Enums/Prediction.cs b/StableDiffusion.NET/Enums/Prediction.cs index f753030..4ccd5b5 100644 --- a/StableDiffusion.NET/Enums/Prediction.cs +++ b/StableDiffusion.NET/Enums/Prediction.cs @@ -2,11 +2,11 @@ public enum Prediction { - Default, EPS, V, EDM_V, - SD3Flow, + Flow, FluxFlow, - Flux2Flow + Flux2Flow, + Default } diff --git a/StableDiffusion.NET/Native/Native.cs b/StableDiffusion.NET/Native/Native.cs index 16ef550..ffdd241 100644 --- a/StableDiffusion.NET/Native/Native.cs +++ b/StableDiffusion.NET/Native/Native.cs @@ -222,8 +222,8 @@ internal unsafe partial class Native [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, 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)] diff --git a/StableDiffusion.NET/StableDiffusionCpp.cs b/StableDiffusion.NET/StableDiffusionCpp.cs index 13e4c0a..7ff65f2 100644 --- a/StableDiffusion.NET/StableDiffusionCpp.cs +++ b/StableDiffusion.NET/StableDiffusionCpp.cs @@ -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 PreprocessCanny(CannyParameter parameter) {