diff --git a/README.md b/README.md index 10f6ba1..131d453 100644 --- a/README.md +++ b/README.md @@ -12,5 +12,5 @@ Run `build.bat` to build the native libs (modify params like CUDA-builds if need ```csharp using StableDiffusionModel sd = new(@"", new ModelParameter()); using StableDiffusionParameter parameter = new StableDiffusionParameter(); -using Image image = sd.TextToImage("", parameter); +using StableDiffusionImage image = sd.TextToImage("", parameter); ``` diff --git a/StableDiffusion.NET/StableDiffusion.cs b/StableDiffusion.NET/StableDiffusion.cs index a56ff9c..026c6c8 100644 --- a/StableDiffusion.NET/StableDiffusion.cs +++ b/StableDiffusion.NET/StableDiffusion.cs @@ -41,22 +41,22 @@ public sealed unsafe class StableDiffusionModel : IDisposable if (!success) throw new IOException("Failed to load model"); } - public Image TextToImage(string prompt, StableDiffusionParameter parameter) + public StableDiffusionImage 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); + return new StableDiffusionImage(result, parameter.Width, parameter.Height); } - public Image ImageToImage(string prompt, Span image, StableDiffusionParameter parameter) + public StableDiffusionImage ImageToImage(string prompt, Span image, StableDiffusionParameter parameter) { ObjectDisposedException.ThrowIf(_disposed, this); fixed (byte* imagePtr = image) { byte* result = Native.stable_diffusion_image_predict_image(_ctx, parameter.ParamPtr, imagePtr, prompt); - return new Image(result, parameter.Width, parameter.Height); + return new StableDiffusionImage(result, parameter.Width, parameter.Height); } } diff --git a/StableDiffusion.NET/Image.cs b/StableDiffusion.NET/StableDiffusionImage.cs similarity index 83% rename from StableDiffusion.NET/Image.cs rename to StableDiffusion.NET/StableDiffusionImage.cs index 3f4dfbe..5219e31 100644 --- a/StableDiffusion.NET/Image.cs +++ b/StableDiffusion.NET/StableDiffusionImage.cs @@ -2,7 +2,7 @@ namespace StableDiffusion.NET; -public sealed unsafe class Image : IDisposable +public sealed unsafe class StableDiffusionImage : IDisposable { #region Constants @@ -33,14 +33,14 @@ public sealed unsafe class Image : IDisposable #region Constructors - internal Image(byte* ptr, int width, int height) + internal StableDiffusionImage(byte* ptr, int width, int height) { this._imagePtr = ptr; this.Width = width; this.Height = height; } - ~Image() => Dispose(); + ~StableDiffusionImage() => Dispose(); #endregion