Add registration configuration overloads

Author: saddam213

OnnxStack.Core/Registration.cs

@@ -21,6 +21,18 @@ public static void AddOnnxStack(this IServiceCollection serviceCollection)
         }
 
 
+        /// <summary>
+        /// Register OnnxStack services
+        /// </summary>
+        /// <param name="serviceCollection">The service collection.</param>
+        /// <param name="configuration">The configuration.</param>
+        public static void AddOnnxStack(this IServiceCollection serviceCollection, OnnxStackConfig configuration)
+        {
+            serviceCollection.AddSingleton(configuration);
+            serviceCollection.AddSingleton<IOnnxModelService, OnnxModelService>();
+        }
+
+
         /// <summary>
         /// Register a custom IConfigSection section that is in the appsettings.json
         /// </summary>

OnnxStack.StableDiffusion/Registration.cs

@@ -21,9 +21,27 @@ public static class Registration
         /// <param name="serviceCollection">The service collection.</param>
         public static void AddOnnxStackStableDiffusion(this IServiceCollection serviceCollection)
         {
-            ConfigureLibraries();
             serviceCollection.AddOnnxStack();
+            serviceCollection.RegisterServices();
             serviceCollection.AddSingleton(ConfigManager.LoadConfiguration<StableDiffusionConfig>(nameof(OnnxStackConfig)));
+        }
+
+
+        /// <summary>
+        /// Register OnnxStack StableDiffusion services, AddOnnxStack() must be called before
+        /// </summary>
+        /// <param name="serviceCollection">The service collection.</param>
+        /// <param name="configuration">The configuration.</param>
+        public static void AddOnnxStackStableDiffusion(this IServiceCollection serviceCollection, StableDiffusionConfig configuration)
+        {
+            serviceCollection.RegisterServices();
+            serviceCollection.AddSingleton(configuration);
+        }
+
+
+        private static void RegisterServices(this IServiceCollection serviceCollection)
+        {
+            ConfigureLibraries();
 
             // Services
             serviceCollection.AddSingleton<IPromptService, PromptService>();