Merge pull request #19 from drasticactions/load-native-library

Allow manually loading SD library

Author: DarthAffe

StableDiffusion.NET/Native/Native.Load.cs

@@ -26,6 +26,18 @@ static Native()
 
     #region Methods
 
+    internal static bool LoadNativeLibrary(string libraryPath)
+    {
+        if (_loadedLibraryHandle != nint.Zero) return true;
+        if (NativeLibrary.TryLoad(libraryPath, out nint handle))
+        {
+            _loadedLibraryHandle = handle;
+            return true;
+        }
+
+        return false;
+    }
+
     private static nint ResolveDllImport(string libraryname, Assembly assembly, DllImportSearchPath? searchpath)
     {
         if (libraryname != LIB_NAME) return nint.Zero;

StableDiffusion.NET/StableDiffusionModel.cs

@@ -91,6 +91,15 @@ private void Initialize()
         }
     }
 
+    /// <summary>
+    /// Manually load the native stable diffusion library.
+    /// Once set, it will continue to be used for all instances.
+    /// </summary>
+    /// <param name="libraryPath">Path to the stable diffusion library.</param>
+    /// <returns>Bool if the library loaded.</returns>
+    public static bool LoadNativeLibrary(string libraryPath)
+        => Native.LoadNativeLibrary(libraryPath);
+
     public IImage<ColorRGB> TextToImage(string prompt, StableDiffusionParameter parameter)
     {
         ObjectDisposedException.ThrowIf(_disposed, this);