From b188f58777d42884747bc6419e8b01adc8d4b339 Mon Sep 17 00:00:00 2001 From: Konstantin Date: Fri, 20 Jan 2023 19:36:10 +0100 Subject: Bugfix, C# language projection --- WhisperNet/Library.cs | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'WhisperNet/Library.cs') 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 ) )] 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; } /// Load Whisper model on a background thread, with optional progress reporting and cancellation - public static Task loadModelAsync( string path, CancellationToken cancelToken, Action? pfnProgress = null, eModelImplementation impl = eModelImplementation.GPU ) + public static Task loadModelAsync( string path, CancellationToken cancelToken, eGpuModelFlags flags = eGpuModelFlags.None, Action? pfnProgress = null, eModelImplementation impl = eModelImplementation.GPU ) { TaskCompletionSource tcs = new TaskCompletionSource(); @@ -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 ); -- cgit v1.2.3