mirror of
https://github.com/DarthAffe/StableDiffusion.NET.git
synced 2025-12-12 13:28:35 +00:00
Compare commits
7 Commits
a2827a082d
...
b147a7a63e
| Author | SHA1 | Date | |
|---|---|---|---|
| b147a7a63e | |||
| 82f25d1fda | |||
| 8780800766 | |||
| 020a3f02eb | |||
| 7d3e8c301d | |||
| 059e4ec299 | |||
| 4b5d4c9b07 |
62
.github/workflows/backends.yml
vendored
62
.github/workflows/backends.yml
vendored
@ -111,52 +111,6 @@ jobs:
|
||||
name: windows-cuda12
|
||||
path: .\build\bin\Release\stable-diffusion.dll
|
||||
|
||||
windows-sycl:
|
||||
runs-on: windows-latest
|
||||
|
||||
defaults:
|
||||
run:
|
||||
shell: bash
|
||||
|
||||
env:
|
||||
WINDOWS_BASEKIT_URL: https://registrationcenter-download.intel.com/akdlm/IRC_NAS/b380d914-366b-4b77-a74a-05e3c38b3514/intel-oneapi-base-toolkit-2025.0.0.882_offline.exe
|
||||
WINDOWS_DPCPP_MKL: intel.oneapi.win.cpp-dpcpp-common:intel.oneapi.win.mkl.devel:intel.oneapi.win.dnnl:intel.oneapi.win.tbb.devel
|
||||
ONEAPI_ROOT: "C:/Program Files (x86)/Intel/oneAPI"
|
||||
|
||||
steps:
|
||||
- name: Checkout
|
||||
id: checkout
|
||||
uses: actions/checkout@v4.2.2
|
||||
with:
|
||||
repository: 'leejet/stable-diffusion.cpp'
|
||||
ref: '${{ github.event.inputs.commit }}'
|
||||
submodules: recursive
|
||||
|
||||
- name: Checkout Install Script
|
||||
id: checkoutInstallScript
|
||||
uses: actions/checkout@v4.2.2
|
||||
with:
|
||||
path: scripts
|
||||
sparse-checkout: |
|
||||
Backends/install-oneapi.bat
|
||||
Backends/win-build-sycl.bat
|
||||
sparse-checkout-cone-mode: false
|
||||
|
||||
- name: Install Sycl tools
|
||||
id: installSyclCompiler
|
||||
run: scripts/Backends/install-oneapi.bat $WINDOWS_BASEKIT_URL $WINDOWS_DPCPP_MKL
|
||||
|
||||
- name: Build
|
||||
id: cmake_build
|
||||
run: scripts/Backends/win-build-sycl.bat
|
||||
|
||||
- name: Upload artifact
|
||||
id: upload_artifact
|
||||
uses: actions/upload-artifact@v4.6.1
|
||||
with:
|
||||
name: windows-sycl
|
||||
path: .\build\bin\stable-diffusion.dll
|
||||
|
||||
windows-hip:
|
||||
runs-on: windows-latest
|
||||
|
||||
@ -166,14 +120,20 @@ jobs:
|
||||
run: |
|
||||
$ErrorActionPreference = "Stop"
|
||||
write-host "Downloading AMD HIP SDK Installer"
|
||||
Invoke-WebRequest -Uri "https://download.amd.com/developer/eula/rocm-hub/AMD-Software-PRO-Edition-24.Q4-WinSvr2022-For-HIP.exe" -OutFile "${env:RUNNER_TEMP}\rocm-install.exe"
|
||||
Invoke-WebRequest -Uri "https://download.amd.com/developer/eula/rocm-hub/AMD-Software-PRO-Edition-25.Q3-WinSvr2022-For-HIP.exe" -OutFile "${env:RUNNER_TEMP}\rocm-install.exe"
|
||||
write-host "Installing AMD HIP SDK"
|
||||
Start-Process "${env:RUNNER_TEMP}\rocm-install.exe" -ArgumentList '-install' -NoNewWindow -Wait
|
||||
write-host "Completed AMD HIP SDK installation"
|
||||
- name: Verify ROCm
|
||||
id: verify
|
||||
run: |
|
||||
& 'C:\Program Files\AMD\ROCm\*\bin\clang.exe' --version
|
||||
# Find and test ROCm installation
|
||||
$clangPath = Get-ChildItem 'C:\Program Files\AMD\ROCm\*\bin\clang.exe' | Select-Object -First 1
|
||||
if (-not $clangPath) {
|
||||
Write-Error "ROCm installation not found"
|
||||
exit 1
|
||||
}
|
||||
& $clangPath.FullName --version
|
||||
|
||||
- name: Checkout
|
||||
id: checkout
|
||||
@ -204,7 +164,7 @@ jobs:
|
||||
runs-on: windows-latest
|
||||
|
||||
env:
|
||||
VULKAN_VERSION: 1.3.261.1
|
||||
VULKAN_VERSION: 1.4.328.1
|
||||
|
||||
steps:
|
||||
- name: Checkout
|
||||
@ -218,7 +178,7 @@ jobs:
|
||||
- name: Install Vulkan SDK
|
||||
id: get_vulkan
|
||||
run: |
|
||||
curl.exe -o $env:RUNNER_TEMP/VulkanSDK-Installer.exe -L "https://sdk.lunarg.com/sdk/download/${env:VULKAN_VERSION}/windows/VulkanSDK-${env:VULKAN_VERSION}-Installer.exe"
|
||||
curl.exe -o $env:RUNNER_TEMP/VulkanSDK-Installer.exe -L "https://sdk.lunarg.com/sdk/download/${env:VULKAN_VERSION}/windows/vulkansdk-windows-X64-${env:VULKAN_VERSION}.exe"
|
||||
& "$env:RUNNER_TEMP\VulkanSDK-Installer.exe" --accept-licenses --default-answer --confirm-command install
|
||||
Add-Content $env:GITHUB_ENV "VULKAN_SDK=C:\VulkanSDK\${env:VULKAN_VERSION}"
|
||||
Add-Content $env:GITHUB_PATH "C:\VulkanSDK\${env:VULKAN_VERSION}\bin"
|
||||
@ -354,7 +314,7 @@ jobs:
|
||||
|
||||
linux-hip:
|
||||
runs-on: ubuntu-22.04
|
||||
container: rocm/dev-ubuntu-22.04:6.0.2
|
||||
container: rocm/dev-ubuntu-22.04:6.4.4
|
||||
|
||||
steps:
|
||||
- name: Dependencies
|
||||
|
||||
5
.github/workflows/release.yml
vendored
5
.github/workflows/release.yml
vendored
@ -19,9 +19,10 @@ jobs:
|
||||
fetch-depth: 0
|
||||
|
||||
- name: Install .NET Core
|
||||
uses: actions/setup-dotnet@v4
|
||||
uses: actions/setup-dotnet@v5
|
||||
with:
|
||||
dotnet-version: |
|
||||
10.0.x
|
||||
9.0.x
|
||||
8.0.x
|
||||
|
||||
@ -49,7 +50,7 @@ jobs:
|
||||
with:
|
||||
tag_name: ${{ github.event.inputs.version }}
|
||||
generate_release_notes: true
|
||||
files: StableDiffusion.NET/bin/Release/net9.0/StableDiffusion.NET.dll
|
||||
files: StableDiffusion.NET/bin/Release/net10.0/StableDiffusion.NET.dll
|
||||
|
||||
- name: Nuget Push
|
||||
id: nuget_push
|
||||
|
||||
@ -126,6 +126,21 @@ enum sd_log_level_t {
|
||||
SD_LOG_ERROR
|
||||
};
|
||||
|
||||
enum preview_t {
|
||||
PREVIEW_NONE,
|
||||
PREVIEW_PROJ,
|
||||
PREVIEW_TAE,
|
||||
PREVIEW_VAE,
|
||||
PREVIEW_COUNT
|
||||
};
|
||||
|
||||
enum lora_apply_mode_t {
|
||||
LORA_APPLY_AUTO,
|
||||
LORA_APPLY_IMMEDIATELY,
|
||||
LORA_APPLY_AT_RUNTIME,
|
||||
LORA_APPLY_MODE_COUNT,
|
||||
};
|
||||
|
||||
typedef struct {
|
||||
bool enabled;
|
||||
int tile_size_x;
|
||||
@ -157,11 +172,13 @@ typedef struct {
|
||||
enum sd_type_t wtype;
|
||||
enum rng_type_t rng_type;
|
||||
enum prediction_t prediction;
|
||||
enum lora_apply_mode_t lora_apply_mode;
|
||||
bool offload_params_to_cpu;
|
||||
bool keep_clip_on_cpu;
|
||||
bool keep_control_net_on_cpu;
|
||||
bool keep_vae_on_cpu;
|
||||
bool diffusion_flash_attn;
|
||||
bool tae_preview_only;
|
||||
bool diffusion_conv_direct;
|
||||
bool vae_conv_direct;
|
||||
bool force_sdxl_vae_conv_scale;
|
||||
@ -254,9 +271,11 @@ 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);
|
||||
|
||||
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 const char* sd_get_system_info();
|
||||
|
||||
@ -270,6 +289,10 @@ 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_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);
|
||||
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_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);
|
||||
|
||||
8
StableDiffusion.NET/Enums/LoraApplyMode.cs
Normal file
8
StableDiffusion.NET/Enums/LoraApplyMode.cs
Normal file
@ -0,0 +1,8 @@
|
||||
namespace StableDiffusion.NET;
|
||||
|
||||
public enum LoraApplyMode
|
||||
{
|
||||
Auto,
|
||||
Immediately,
|
||||
AtRuntime
|
||||
}
|
||||
9
StableDiffusion.NET/Enums/Preview.cs
Normal file
9
StableDiffusion.NET/Enums/Preview.cs
Normal file
@ -0,0 +1,9 @@
|
||||
namespace StableDiffusion.NET;
|
||||
|
||||
public enum Preview
|
||||
{
|
||||
None,
|
||||
Proj,
|
||||
TAE,
|
||||
VAE
|
||||
}
|
||||
@ -0,0 +1,15 @@
|
||||
using System;
|
||||
using HPPH;
|
||||
|
||||
namespace StableDiffusion.NET;
|
||||
|
||||
public sealed class StableDiffusionPreviewEventArgs(int step, bool isNoisy, Image<ColorRGB> image) : EventArgs
|
||||
{
|
||||
#region Properties & Fields
|
||||
|
||||
public int Step { get; } = step;
|
||||
public bool IsNoisy { get; } = isNoisy;
|
||||
public Image<ColorRGB> Image { get; } = image;
|
||||
|
||||
#endregion
|
||||
}
|
||||
@ -72,6 +72,8 @@ public sealed class DiffusionModelParameter
|
||||
/// </summary>
|
||||
public bool FlashAttention { get; set; } = false;
|
||||
|
||||
public bool TaePreviewOnly { get; set; } = false;
|
||||
|
||||
/// <summary>
|
||||
/// use Conv2d direct in the diffusion model
|
||||
/// This might crash if it is not supported by the backend.
|
||||
@ -91,6 +93,8 @@ public sealed class DiffusionModelParameter
|
||||
|
||||
public Prediction Prediction { get; set; } = Prediction.Default;
|
||||
|
||||
public LoraApplyMode LoraApplyMode { get; set; } = LoraApplyMode.Auto;
|
||||
|
||||
/// <summary>
|
||||
/// quantizes on load
|
||||
/// not really useful in most cases
|
||||
|
||||
@ -30,11 +30,13 @@ internal static unsafe class DiffusionModelParameterMarshaller
|
||||
wtype = managed.Quantization,
|
||||
rng_type = managed.RngType,
|
||||
prediction = managed.Prediction,
|
||||
lora_apply_mode = managed.LoraApplyMode,
|
||||
offload_params_to_cpu = (sbyte)(managed.OffloadParamsToCPU ? 1 : 0),
|
||||
keep_clip_on_cpu = (sbyte)(managed.KeepClipOnCPU ? 1 : 0),
|
||||
keep_control_net_on_cpu = (sbyte)(managed.KeepControlNetOnCPU ? 1 : 0),
|
||||
keep_vae_on_cpu = (sbyte)(managed.KeepVaeOnCPU ? 1 : 0),
|
||||
diffusion_flash_attn = (sbyte)(managed.FlashAttention ? 1 : 0),
|
||||
tae_preview_only = (sbyte)(managed.TaePreviewOnly ? 1 : 0),
|
||||
diffusion_conv_direct = (sbyte)(managed.DiffusionConvDirect ? 1 : 0),
|
||||
vae_conv_direct = (sbyte)(managed.VaeConvDirect ? 1 : 0),
|
||||
force_sdxl_vae_conv_scale = (sbyte)(managed.ForceSdxlVaeConvScale ? 1 : 0),
|
||||
@ -68,11 +70,13 @@ internal static unsafe class DiffusionModelParameterMarshaller
|
||||
Quantization = unmanaged.wtype,
|
||||
RngType = unmanaged.rng_type,
|
||||
Prediction = unmanaged.prediction,
|
||||
LoraApplyMode = unmanaged.lora_apply_mode,
|
||||
OffloadParamsToCPU = unmanaged.offload_params_to_cpu == 1,
|
||||
KeepClipOnCPU = unmanaged.keep_clip_on_cpu == 1,
|
||||
KeepControlNetOnCPU = unmanaged.keep_control_net_on_cpu == 1,
|
||||
KeepVaeOnCPU = unmanaged.keep_vae_on_cpu == 1,
|
||||
FlashAttention = unmanaged.diffusion_flash_attn == 1,
|
||||
TaePreviewOnly = unmanaged.tae_preview_only == 1,
|
||||
DiffusionConvDirect = unmanaged.diffusion_conv_direct == 1,
|
||||
VaeConvDirect = unmanaged.vae_conv_direct == 1,
|
||||
ForceSdxlVaeConvScale = unmanaged.force_sdxl_vae_conv_scale == 1,
|
||||
|
||||
@ -23,6 +23,8 @@ using sd_img_gen_params_t = ImageGenerationParameter;
|
||||
using sd_log_level_t = LogLevel;
|
||||
using sd_type_t = Quantization;
|
||||
using sd_vid_gen_params_t = VideoGenerationParameter;
|
||||
using lora_apply_mode_t = LoraApplyMode;
|
||||
using preview_t = Preview;
|
||||
using size_t = nuint;
|
||||
using uint32_t = uint;
|
||||
using uint8_t = byte;
|
||||
@ -73,11 +75,13 @@ internal unsafe partial class Native
|
||||
public sd_type_t wtype;
|
||||
public rng_type_t rng_type;
|
||||
public prediction_t prediction;
|
||||
public lora_apply_mode_t lora_apply_mode;
|
||||
public sbyte offload_params_to_cpu;
|
||||
public sbyte keep_clip_on_cpu;
|
||||
public sbyte keep_control_net_on_cpu;
|
||||
public sbyte keep_vae_on_cpu;
|
||||
public sbyte diffusion_flash_attn;
|
||||
public sbyte tae_preview_only;
|
||||
public sbyte diffusion_conv_direct;
|
||||
public sbyte vae_conv_direct;
|
||||
public sbyte force_sdxl_vae_conv_scale;
|
||||
@ -188,6 +192,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);
|
||||
|
||||
#endregion
|
||||
|
||||
@ -199,6 +204,9 @@ internal unsafe partial class Native
|
||||
[LibraryImport(LIB_NAME, EntryPoint = "sd_set_progress_callback")]
|
||||
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);
|
||||
|
||||
[LibraryImport(LIB_NAME, EntryPoint = "get_num_physical_cores")]
|
||||
internal static partial int32_t get_num_physical_cores();
|
||||
|
||||
@ -243,6 +251,20 @@ internal unsafe partial class Native
|
||||
[LibraryImport(LIB_NAME, EntryPoint = "str_to_prediction")]
|
||||
internal static partial prediction_t str_to_prediction([MarshalAs(UnmanagedType.LPStr)] string str);
|
||||
|
||||
[LibraryImport(LIB_NAME, EntryPoint = "sd_preview_name")]
|
||||
[return: MarshalAs(UnmanagedType.LPStr)]
|
||||
internal static partial string sd_preview_name(preview_t preview);
|
||||
|
||||
[LibraryImport(LIB_NAME, EntryPoint = "str_to_preview")]
|
||||
internal static partial preview_t str_to_preview([MarshalAs(UnmanagedType.LPStr)] string str);
|
||||
|
||||
[LibraryImport(LIB_NAME, EntryPoint = "sd_lora_apply_mode_name")]
|
||||
[return: MarshalAs(UnmanagedType.LPStr)]
|
||||
internal static partial string sd_lora_apply_mode_name(lora_apply_mode_t mode);
|
||||
|
||||
[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_ctx_params_init")]
|
||||
|
||||
@ -1,7 +1,7 @@
|
||||
<Project Sdk="Microsoft.NET.Sdk">
|
||||
|
||||
<PropertyGroup>
|
||||
<TargetFrameworks>net9.0;net8.0</TargetFrameworks>
|
||||
<TargetFrameworks>net10.0;net9.0;net8.0</TargetFrameworks>
|
||||
<Nullable>enable</Nullable>
|
||||
<AllowUnsafeBlocks>true</AllowUnsafeBlocks>
|
||||
|
||||
|
||||
@ -1,6 +1,6 @@
|
||||
using System;
|
||||
using HPPH;
|
||||
using HPPH;
|
||||
using JetBrains.Annotations;
|
||||
using System;
|
||||
|
||||
namespace StableDiffusion.NET;
|
||||
|
||||
@ -12,6 +12,7 @@ public static unsafe class StableDiffusionCpp
|
||||
// ReSharper disable NotAccessedField.Local - They are important, the delegate can be collected if it's not stored!
|
||||
private static Native.sd_log_cb_t? _logCallback;
|
||||
private static Native.sd_progress_cb_t? _progressCallback;
|
||||
private static Native.sd_preview_cb_t? _previewCallback;
|
||||
// ReSharper restore NotAccessedField.Local
|
||||
|
||||
#endregion
|
||||
@ -20,6 +21,7 @@ public static unsafe class StableDiffusionCpp
|
||||
|
||||
public static event EventHandler<StableDiffusionLogEventArgs>? Log;
|
||||
public static event EventHandler<StableDiffusionProgressEventArgs>? Progress;
|
||||
public static event EventHandler<StableDiffusionPreviewEventArgs>? Preview;
|
||||
|
||||
#endregion
|
||||
|
||||
@ -33,6 +35,19 @@ public static unsafe class StableDiffusionCpp
|
||||
Native.sd_set_progress_callback(_progressCallback = OnNativeProgress, null);
|
||||
}
|
||||
|
||||
public static void EnablePreview(Preview mode, int interval, bool denoised, bool noisy)
|
||||
{
|
||||
ArgumentOutOfRangeException.ThrowIfNegative(interval);
|
||||
|
||||
if (mode == NET.Preview.None)
|
||||
_previewCallback = null;
|
||||
|
||||
else if (_previewCallback == null)
|
||||
_previewCallback = OnPreview;
|
||||
|
||||
Native.sd_set_preview_callback(_previewCallback, mode, interval, denoised, noisy);
|
||||
}
|
||||
|
||||
public static void Convert(string modelPath, string vaePath, Quantization quantization, string outputPath, string tensorTypeRules = "")
|
||||
{
|
||||
ArgumentException.ThrowIfNullOrWhiteSpace(nameof(modelPath));
|
||||
@ -89,5 +104,18 @@ public static unsafe class StableDiffusionCpp
|
||||
catch { /**/ }
|
||||
}
|
||||
|
||||
private static void OnPreview(int step, int frameCount, Native.Types.sd_image_t* frames, bool isNoisy)
|
||||
{
|
||||
try
|
||||
{
|
||||
if (frameCount <= 0 || frames == null) return;
|
||||
|
||||
Image<ColorRGB> image = ImageHelper.GetImage(frames, 0);
|
||||
|
||||
Preview?.Invoke(null, new StableDiffusionPreviewEventArgs(step, isNoisy, image));
|
||||
}
|
||||
catch { /**/ }
|
||||
}
|
||||
|
||||
#endregion
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user