@remotion/install-whisper-cpp 4.0.322 → 4.0.324

package/dist/transcribe.d.ts

@@ -51,6 +51,7 @@ type Params = {
 type Result = {
     language: string;
 };
+type AdditionalArgs = string[] | [string, string][];
 export type TranscriptionJson<WithTokenLevelTimestamp extends boolean> = {
     systeminfo: string;
     model: Model;
@@ -60,7 +61,7 @@ export type TranscriptionJson<WithTokenLevelTimestamp extends boolean> = {
 };
 export type TranscribeOnProgress = (progress: number) => void;
 export declare const modelToDtw: (model: WhisperModel) => string;
-export declare const transcribe: <HasTokenLevelTimestamps extends boolean>({ inputPath, whisperPath, whisperCppVersion, model, modelFolder, translateToEnglish, tokenLevelTimestamps, printOutput, tokensPerItem, language, splitOnWord, signal, onProgress, }: {
+export declare const transcribe: <HasTokenLevelTimestamps extends boolean>({ inputPath, whisperPath, whisperCppVersion, model, modelFolder, translateToEnglish, tokenLevelTimestamps, printOutput, tokensPerItem, language, splitOnWord, signal, onProgress, flashAttention, additionalArgs, }: {
     inputPath: string;
     whisperPath: string;
     whisperCppVersion: string;
@@ -74,5 +75,7 @@ export declare const transcribe: <HasTokenLevelTimestamps extends boolean>({ inp
     splitOnWord?: boolean;
     signal?: AbortSignal;
     onProgress?: TranscribeOnProgress;
+    flashAttention?: boolean;
+    additionalArgs?: AdditionalArgs;
 }) => Promise<TranscriptionJson<HasTokenLevelTimestamps>>;
 export {};

package/dist/transcribe.js

@@ -71,7 +71,7 @@ const modelToDtw = (model) => {
     return model;
 };
 exports.modelToDtw = modelToDtw;
-const transcribeToTemporaryFile = async ({ fileToTranscribe, whisperPath, whisperCppVersion, model, tmpJSONPath, modelFolder, translate, tokenLevelTimestamps, printOutput, tokensPerItem, language, splitOnWord, signal, onProgress, }) => {
+const transcribeToTemporaryFile = async ({ fileToTranscribe, whisperPath, whisperCppVersion, model, tmpJSONPath, modelFolder, translate, tokenLevelTimestamps, printOutput, tokensPerItem, language, splitOnWord, signal, onProgress, flashAttention, additionalArgs, }) => {
     const modelPath = (0, download_whisper_model_1.getModelPath)(modelFolder !== null && modelFolder !== void 0 ? modelFolder : whisperPath, model);
     if (!node_fs_1.default.existsSync(modelPath)) {
         throw new Error(`Error: Model ${model} does not exist at ${modelFolder ? modelFolder : modelPath}. Check out the downloadWhisperModel() API at https://www.remotion.dev/docs/install-whisper-cpp/download-whisper-model to see how to install whisper models`);
@@ -91,6 +91,8 @@ const transcribeToTemporaryFile = async ({ fileToTranscribe, whisperPath, whispe
         translate ? '-tr' : null,
         language ? ['-l', language.toLowerCase()] : null,
         splitOnWord ? ['--split-on-word', splitOnWord] : null,
+        flashAttention ? ['--flash-attn', 'true'] : null,
+        ...(additionalArgs !== null && additionalArgs !== void 0 ? additionalArgs : []),
     ]
         .flat(1)
         .filter(Boolean);
@@ -151,7 +153,7 @@ const transcribeToTemporaryFile = async ({ fileToTranscribe, whisperPath, whispe
     });
     return { outputPath };
 };
-const transcribe = async ({ inputPath, whisperPath, whisperCppVersion, model, modelFolder, translateToEnglish = false, tokenLevelTimestamps, printOutput = true, tokensPerItem, language, splitOnWord, signal, onProgress, }) => {
+const transcribe = async ({ inputPath, whisperPath, whisperCppVersion, model, modelFolder, translateToEnglish = false, tokenLevelTimestamps, printOutput = true, tokensPerItem, language, splitOnWord, signal, onProgress, flashAttention, additionalArgs, }) => {
     if (!(0, node_fs_1.existsSync)(whisperPath)) {
         throw new Error(`Whisper does not exist at ${whisperPath}. Double-check the passed whisperPath. If you havent installed whisper, check out the installWhisperCpp() API at https://www.remotion.dev/docs/install-whisper-cpp/install-whisper-cpp to see how to install whisper programatically.`);
     }
@@ -177,6 +179,8 @@ const transcribe = async ({ inputPath, whisperPath, whisperCppVersion, model, mo
         signal: signal !== null && signal !== void 0 ? signal : null,
         splitOnWord: splitOnWord !== null && splitOnWord !== void 0 ? splitOnWord : null,
         onProgress: onProgress !== null && onProgress !== void 0 ? onProgress : null,
+        flashAttention,
+        additionalArgs,
     });
     const json = (await readJson(tmpJSONPath));
     node_fs_1.default.unlinkSync(tmpJSONPath);

package/package.json

@@ -3,7 +3,7 @@
     "url": "https://github.com/remotion-dev/remotion/tree/main/packages/install-whisper-cpp"
   },
   "name": "@remotion/install-whisper-cpp",
-  "version": "4.0.322",
+  "version": "4.0.324",
   "description": "Helpers for installing and using Whisper.cpp",
   "main": "dist/index.js",
   "sideEffects": false,
@@ -16,12 +16,12 @@
   "author": "Jonny Burger <jonny@remotion.dev>",
   "license": "SEE LICENSE IN LICENSE.md",
   "dependencies": {
-    "@remotion/captions": "4.0.322"
+    "@remotion/captions": "4.0.324"
   },
   "peerDependencies": {},
   "devDependencies": {
     "eslint": "9.19.0",
-    "@remotion/eslint-config-internal": "4.0.322"
+    "@remotion/eslint-config-internal": "4.0.324"
   },
   "keywords": [
     "remotion",