UNPKG

smart-whisper-electron

Version:

Whisper.cpp Node.js binding with auto model offloading strategy.

117 lines (115 loc) 3.87 kB
"use strict"; var __create = Object.create; var __defProp = Object.defineProperty; var __getOwnPropDesc = Object.getOwnPropertyDescriptor; var __getOwnPropNames = Object.getOwnPropertyNames; var __getProtoOf = Object.getPrototypeOf; var __hasOwnProp = Object.prototype.hasOwnProperty; var __export = (target, all) => { for (var name in all) __defProp(target, name, { get: all[name], enumerable: true }); }; var __copyProps = (to, from, except, desc) => { if (from && typeof from === "object" || typeof from === "function") { for (let key of __getOwnPropNames(from)) if (!__hasOwnProp.call(to, key) && key !== except) __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable }); } return to; }; var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps( // If the importer is in node compatibility mode or this is not an ESM // file that has been converted to a CommonJS file using a Babel- // compatible transform (i.e. "__esModule" has not been set), then set // "default" to the CommonJS "module.exports" for node compatibility. isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target, mod )); var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod); // src/build.ts var build_exports = {}; __export(build_exports, { defines: () => defines, libraries: () => libraries, sources: () => sources }); module.exports = __toCommonJS(build_exports); var import_node_os = __toESM(require("os")); var import_node_child_process = require("child_process"); var cfg = config(); var sources = cfg.sources.join(" "); var defines = cfg.defines.join(" "); var libraries = cfg.libraries.join(" "); function config() { var _a; if (process.env.BYOL) { return { sources: [], defines: [], libraries: [process.env.BYOL] }; } const COMPUTE_BACKEND = (_a = process.env.COMPUTE_BACKEND) != null ? _a : infer_backend(); const cfg2 = { sources: [ "whisper.cpp/src/whisper.cpp", "whisper.cpp/ggml/src/ggml.c", "whisper.cpp/ggml/src/ggml-alloc.c", "whisper.cpp/ggml/src/ggml-backend.c", "whisper.cpp/ggml/src/ggml-quants.c", "whisper.cpp/ggml/src/ggml-aarch64.c" ], defines: [], libraries: [] }; switch (COMPUTE_BACKEND) { case "accelerate": { cfg2.defines.push("GGML_USE_ACCELERATE"); cfg2.libraries.push('"-framework Foundation"'); cfg2.libraries.push('"-framework Accelerate"'); break; } case "metal": { cfg2.sources.push("whisper.cpp/ggml/src/ggml-metal.m"); cfg2.defines.push("GGML_USE_ACCELERATE"); cfg2.defines.push("GGML_USE_METAL"); cfg2.libraries.push('"-framework Foundation"'); cfg2.libraries.push('"-framework Accelerate"'); cfg2.libraries.push('"-framework Metal"'); cfg2.libraries.push('"-framework MetalKit"'); break; } case "openblas": { cfg2.defines.push("GGML_USE_OPENBLAS"); cfg2.libraries.push("-lopenblas"); break; } default: { } } return cfg2; } function infer_backend() { let backend = "cpu"; try { if (import_node_os.default.platform() === "darwin") { backend = "accelerate"; if (import_node_os.default.arch() === "arm64") { backend = "metal"; } } else if (import_node_os.default.platform() === "linux") { const has_libopenblas = !!(0, import_node_child_process.execSync)("ldconfig -p | grep libopenblas").toString().trim(); if (has_libopenblas) { backend = "openblas"; } } } catch (e) { } return backend; } // Annotate the CommonJS export names for ESM import in node: 0 && (module.exports = { defines, libraries, sources });