summaryrefslogtreecommitdiffstats
path: root/WhisperNet/Library.cs
diff options
context:
space:
mode:
authorKonstantin <const@const.me>2023-01-20 19:36:10 +0100
committerKonstantin <const@const.me>2023-01-20 19:36:10 +0100
commitb188f58777d42884747bc6419e8b01adc8d4b339 (patch)
tree0b4cf1f26e4ba9f0f2c4024df7ed5a493da7d034 /WhisperNet/Library.cs
parent64d61ad2726d4c9cf822c7ad3475bf3095bd6583 (diff)
Bugfix, C# language projection
Diffstat (limited to 'WhisperNet/Library.cs')
-rw-r--r--WhisperNet/Library.cs8
1 files changed, 4 insertions, 4 deletions
diff --git a/WhisperNet/Library.cs b/WhisperNet/Library.cs
index 5bdb0a3..ef10666 100644
--- a/WhisperNet/Library.cs
+++ b/WhisperNet/Library.cs
@@ -27,7 +27,7 @@ namespace Whisper
internal static extern void setupLogger( [In] ref sLoggerSetup setup );
[DllImport( dll, CallingConvention = RuntimeClass.defaultCallingConvention, PreserveSig = true )]
- static extern int loadModel( [MarshalAs( UnmanagedType.LPWStr )] string path, eModelImplementation impl,
+ static extern int loadModel( [MarshalAs( UnmanagedType.LPWStr )] string path, eModelImplementation impl, eGpuModelFlags flags,
[In] ref sLoadModelCallbacks callbacks,
[MarshalAs( UnmanagedType.CustomMarshaler, MarshalTypeRef = typeof( Marshaler<iModel> ) )] out iModel model );
@@ -40,13 +40,13 @@ namespace Whisper
iModel model;
sLoadModelCallbacks callbacks = default;
NativeLogger.prologue();
- int hr = loadModel( path, impl, ref callbacks, out model );
+ int hr = loadModel( path, impl, flags, ref callbacks, out model );
NativeLogger.throwForHR( hr );
return model;
}
/// <summary>Load Whisper model on a background thread, with optional progress reporting and cancellation</summary>
- public static Task<iModel> loadModelAsync( string path, CancellationToken cancelToken, Action<double>? pfnProgress = null, eModelImplementation impl = eModelImplementation.GPU )
+ public static Task<iModel> loadModelAsync( string path, CancellationToken cancelToken, eGpuModelFlags flags = eGpuModelFlags.None, Action<double>? pfnProgress = null, eModelImplementation impl = eModelImplementation.GPU )
{
TaskCompletionSource<iModel> tcs = new TaskCompletionSource<iModel>();
@@ -58,7 +58,7 @@ namespace Whisper
iModel model;
NativeLogger.prologue();
- int hr = loadModel( path, impl, ref callbacks, out model );
+ int hr = loadModel( path, impl, flags, ref callbacks, out model );
NativeLogger.throwForHR( hr );
tcs.SetResult( model );