Merge pull request #33 from DarthAffe/UpscalerFix

Fixed wrong free of upscaled images

Author: DarthAffe

StableDiffusion.NET/Helper/ImageHelper.cs

@@ -8,21 +8,29 @@ internal static class ImageHelper
 {
     public static unsafe Image<ColorRGB> ToImage(Native.sd_image_t* sdImage)
     {
-        int width = (int)sdImage->width;
-        int height = (int)sdImage->height;
-        int bpp = (int)sdImage->channel;
+        Image<ColorRGB> image = ToImage(*sdImage);
 
-        Image<ColorRGB> image = Image<ColorRGB>.Create(new ReadOnlySpan<byte>(sdImage->data, width * height * bpp), width, height, width * bpp);
+        Marshal.FreeHGlobal((nint)sdImage);
+
+        return image;
+    }
+
+    public static unsafe Image<ColorRGB> ToImage(Native.sd_image_t sdImage)
+    {
+        int width = (int)sdImage.width;
+        int height = (int)sdImage.height;
+        int bpp = (int)sdImage.channel;
+
+        Image<ColorRGB> image = Image<ColorRGB>.Create(new ReadOnlySpan<byte>(sdImage.data, width * height * bpp), width, height, width * bpp);
 
         Dispose(sdImage);
 
         return image;
     }
 
-    public static unsafe void Dispose(Native.sd_image_t* image)
+    public static unsafe void Dispose(Native.sd_image_t image)
     {
-        Marshal.FreeHGlobal((nint)image->data);
-        Marshal.FreeHGlobal((nint)image);
+        Marshal.FreeHGlobal((nint)image.data);
     }
 
     public static unsafe Native.sd_image_t ToSdImage(this IImage<ColorRGB> image, byte* pinnedReference)

StableDiffusion.NET/Models/UpscaleModel.cs

@@ -58,14 +58,14 @@ public IImage<ColorRGB> Upscale(IImage image, int upscaleFactor)
         fixed (byte* imagePtr = sourceImage.AsRefImage())
         {
             Native.sd_image_t result = Native.upscale(_ctx, sourceImage.ToSdImage(imagePtr), upscaleFactor);
-            return ImageHelper.ToImage(&result);
+            return ImageHelper.ToImage(result);
         }
     }
 
     private IImage<ColorRGB> Upscale(Native.sd_image_t image, int upscaleFactor)
     {
         Native.sd_image_t result = Native.upscale(_ctx, image, upscaleFactor);
-        return ImageHelper.ToImage(&result);
+        return ImageHelper.ToImage(result);
     }
 
     public void Dispose()