Allocate a larger memory buffer for image processing

Author: saddam213

OnnxStack.StableDiffusion/Registration.cs

@@ -1,6 +1,8 @@
 using Microsoft.Extensions.DependencyInjection;
 using OnnxStack.StableDiffusion.Common;
 using OnnxStack.StableDiffusion.Services;
+using SixLabors.ImageSharp;
+using SixLabors.ImageSharp.Memory;
 
 namespace OnnxStack.Core
 {
@@ -15,10 +17,25 @@ public static class Registration
         /// <param name="serviceCollection">The service collection.</param>
         public static void AddOnnxStackStableDiffusion(this IServiceCollection serviceCollection)
         {
+            ConfigureLibraries();
             serviceCollection.AddOnnxStack();
             serviceCollection.AddSingleton<IPromptService, PromptService>();
             serviceCollection.AddSingleton<ISchedulerService, SchedulerService>();
             serviceCollection.AddSingleton<IStableDiffusionService, StableDiffusionService>();
         }
+
+
+        /// <summary>
+        /// Configures any 3rd party libraries.
+        /// </summary>
+        private static void ConfigureLibraries()
+        {
+            // Create a 100MB image buffer pool
+            Configuration.Default.PreferContiguousImageBuffers = true;
+            Configuration.Default.MemoryAllocator = MemoryAllocator.Create(new MemoryAllocatorOptions
+            {
+                MaximumPoolSizeMegabytes = 100,
+            });
+        }
     }
 }