DarthAffe/StableDiffusion.NET
1
0
Fork 0
mirror of https://github.com/DarthAffe/StableDiffusion.NET.git synced 2025-12-12 21:38:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improved handling of disposed stuff

Browse Source
This commit is contained in:
Darth Affe 2023-12-15 10:21:38 +01:00
parent 6cd523d6ee
commit 1c01ab0dae
3 changed files with 46 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
StableDiffusion.NET/Image.cs
View File

@ -12,12 +12,22 @@ public sealed unsafe class Image : IDisposable
#region Properties & Fields
private bool _disposed;
private readonly byte* _imagePtr;
public int Width { get; }
public int Height { get; }
public ReadOnlySpan<byte> Data => new(_imagePtr, Width * Height * BPP);
public ReadOnlySpan<byte> Data
{
get
{
ObjectDisposedException.ThrowIf(_disposed, this);
return new ReadOnlySpan<byte>(_imagePtr, Width * Height * BPP);
}
}
#endregion
@ -38,9 +48,12 @@ public sealed unsafe class Image : IDisposable
public void Dispose()
{
if (_disposed) return;
Native.stable_diffusion_free_buffer(_imagePtr);
GC.SuppressFinalize(this);
_disposed = true;
}
#endregion

9
StableDiffusion.NET/StableDiffusion.cs
View File

@ -7,6 +7,8 @@ public sealed unsafe class StableDiffusionModel : IDisposable
{
#region Properties & Fields
private bool _disposed;
private readonly string _modelPath;
private readonly ModelParameter _parameter;
@ -41,12 +43,16 @@ public sealed unsafe class StableDiffusionModel : IDisposable
public Image TextToImage(string prompt, StableDiffusionParameter parameter)
{
ObjectDisposedException.ThrowIf(_disposed, this);
byte* result = Native.stable_diffusion_predict_image(_ctx, parameter.ParamPtr, prompt);
return new Image(result, parameter.Width, parameter.Height);
}
public Image ImageToImage(string prompt, Span<byte> image, StableDiffusionParameter parameter)
{
ObjectDisposedException.ThrowIf(_disposed, this);
fixed (byte* imagePtr = image)
{
byte* result = Native.stable_diffusion_image_predict_image(_ctx, parameter.ParamPtr, imagePtr, prompt);
@ -56,9 +62,12 @@ public sealed unsafe class StableDiffusionModel : IDisposable
public void Dispose()
{
if (_disposed) return;
Native.stable_diffusion_free(_ctx);
GC.SuppressFinalize(this);
_disposed = true;
}
public static string GetSystemInfo() => Native.stable_diffusion_get_system_info();

23
StableDiffusion.NET/StableDiffusionParameter.cs
View File

@ -6,6 +6,8 @@ public sealed unsafe class StableDiffusionParameter : IDisposable
{
#region Properties & Fields
private bool _disposed;
#pragma warning disable CS8500 // This takes the address of, gets the size of, or declares a pointer to a managed type
internal readonly Native.stable_diffusion_full_params* ParamPtr;
#pragma warning restore CS8500 // This takes the address of, gets the size of, or declares a pointer to a managed type
@ -16,6 +18,8 @@ public sealed unsafe class StableDiffusionParameter : IDisposable
get => _negativePrompt;
set
{
ObjectDisposedException.ThrowIf(_disposed, this);
_negativePrompt = value;
Native.stable_diffusion_full_params_set_negative_prompt(ParamPtr, _negativePrompt);
}
@ -27,6 +31,8 @@ public sealed unsafe class StableDiffusionParameter : IDisposable
get => _cfgScale;
set
{
ObjectDisposedException.ThrowIf(_disposed, this);
_cfgScale = value;
Native.stable_diffusion_full_params_set_cfg_scale(ParamPtr, _cfgScale);
}
@ -38,6 +44,8 @@ public sealed unsafe class StableDiffusionParameter : IDisposable
get => _width;
set
{
ObjectDisposedException.ThrowIf(_disposed, this);
_width = value;
Native.stable_diffusion_full_params_set_width(ParamPtr, _width);
}
@ -49,6 +57,8 @@ public sealed unsafe class StableDiffusionParameter : IDisposable
get => _height;
set
{
ObjectDisposedException.ThrowIf(_disposed, this);
_height = value;
Native.stable_diffusion_full_params_set_height(ParamPtr, _height);
}
@ -60,6 +70,8 @@ public sealed unsafe class StableDiffusionParameter : IDisposable
get => _sampleMethod;
set
{
ObjectDisposedException.ThrowIf(_disposed, this);
_sampleMethod = value;
Native.stable_diffusion_full_params_set_sample_method(ParamPtr, _sampleMethod.GetNativeName() ?? "EULER_A");
}
@ -71,6 +83,8 @@ public sealed unsafe class StableDiffusionParameter : IDisposable
get => _sampleSteps;
set
{
ObjectDisposedException.ThrowIf(_disposed, this);
_sampleSteps = value;
Native.stable_diffusion_full_params_set_sample_steps(ParamPtr, _sampleSteps);
}
@ -82,6 +96,8 @@ public sealed unsafe class StableDiffusionParameter : IDisposable
get => _seed;
set
{
ObjectDisposedException.ThrowIf(_disposed, this);
_seed = value;
Native.stable_diffusion_full_params_set_seed(ParamPtr, _seed);
}
@ -93,6 +109,8 @@ public sealed unsafe class StableDiffusionParameter : IDisposable
get => _batchCount;
set
{
ObjectDisposedException.ThrowIf(_disposed, this);
_batchCount = value;
Native.stable_diffusion_full_params_set_batch_count(ParamPtr, _batchCount);
}
@ -104,6 +122,8 @@ public sealed unsafe class StableDiffusionParameter : IDisposable
get => _strength;
set
{
ObjectDisposedException.ThrowIf(_disposed, this);
_strength = value;
Native.stable_diffusion_full_params_set_strength(ParamPtr, _strength);
}
@ -138,9 +158,12 @@ public sealed unsafe class StableDiffusionParameter : IDisposable
public void Dispose()
{
if (_disposed) return;
Native.stable_diffusion_free_full_params(ParamPtr);
GC.SuppressFinalize(this);
_disposed = true;
}
#endregion