@mlx-node/privacy 0.0.0 → 0.0.8

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mlx-node/privacy",
-  "version": "0.0.0",
+  "version": "0.0.8",
   "homepage": "https://github.com/mlx-node/mlx-node",
   "bugs": {
     "url": "https://github.com/mlx-node/mlx-node/issues"
@@ -28,7 +28,7 @@
     "test": "vite test run"
   },
   "dependencies": {
-    "@mlx-node/core": "0.0.7"
+    "@mlx-node/core": "0.0.8"
   },
   "devDependencies": {
     "@types/node": "^26.0.0"

package/dist/train.d.ts

@@ -1,84 +0,0 @@
-/**
- * Configuration for {@link PrivacyLoraTrainer.create}.
- *
- * All tunables are optional; omitting a field falls back to the
- * documented defaults from the native trainer:
- *
- * - `rank` = 16, `alpha` = 32, `dropout` = 0.05
- * - `loraLr` = 1e-4, `classifierLr` = 5e-5
- * - `batchSize` = 2, `maxSeqLen` = 256, `numEpochs` = 3
- * - `gradAccumSteps` = 4, `gradClip` = 1.0
- * - `saveEvery` = 500 (set 0 to disable intermediate checkpoints)
- * - `padTokenId` = 0
- *
- * `modelPath`, `dataPath`, and `outputDir` are required.
- *
- * The field names match the native NAPI casing
- * ({@link import('@mlx-node/core').PrivacyLoraTrainConfigJs}) so the
- * wrapper can pass values straight through without re-keying.
- */
-export interface PrivacyLoraTrainConfig {
-    /** Filesystem path to the base privacy-filter checkpoint directory. */
-    modelPath: string;
-    /** Path to the training JSONL dataset (pre-tokenized). */
-    dataPath: string;
-    /**
-     * Optional evaluation JSONL dataset path. Currently ignored by the
-     * trainer — reserved for future eval-loop integration.
-     */
-    evalPath?: string;
-    /**
-     * Directory where `adapter.safetensors`, `adapter_config.json`,
-     * `optimizer_state.safetensors`, and `trainer_state.json` are written.
-     */
-    outputDir: string;
-    rank?: number;
-    alpha?: number;
-    dropout?: number;
-    loraLr?: number;
-    classifierLr?: number;
-    batchSize?: number;
-    maxSeqLen?: number;
-    numEpochs?: number;
-    gradAccumSteps?: number;
-    gradClip?: number;
-    saveEvery?: number;
-    resumeFrom?: string;
-    padTokenId?: number;
-}
-/**
- * High-level wrapper around the native `PrivacyLoraTrainerJs` NAPI class.
- *
- * The native binding's `create` factory is synchronous (it loads the base
- * checkpoint, attaches zero-B LoRA adapters, and parses the JSONL dataset
- * inline), but the public API here is intentionally `async` so we can later
- * move work to a worker thread without a breaking change to consumers.
- *
- * `train()` and `saveAdapter()` are already async on the native side — they
- * funnel through `spawn_blocking` so the Node.js event loop stays responsive
- * for the duration of the training run.
- */
-export declare class PrivacyLoraTrainer {
-    private readonly inner;
-    private constructor();
-    /**
-     * Construct a new trainer.
-     *
-     * Loads the base privacy-filter model from `config.modelPath`, attaches
-     * zero-B LoRA adapters on every attention projection, and parses the
-     * pre-tokenized JSONL dataset at `config.dataPath`.
-     */
-    static create(config: PrivacyLoraTrainConfig): Promise<PrivacyLoraTrainer>;
-    /**
-     * Run the full training loop. Resolves with the final optimizer step
-     * count once `numEpochs` epochs have completed.
-     */
-    train(): Promise<number>;
-    /**
-     * Persist adapter weights, optimizer state, and trainer progress to the
-     * configured `outputDir`. Overwrites any previous checkpoint at that
-     * path.
-     */
-    saveAdapter(): Promise<void>;
-}
-//# sourceMappingURL=train.d.ts.map

package/dist/train.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"train.d.ts","sourceRoot":"","sources":["../src/train.ts"],"names":[],"mappings":"AAKA;;;;;;;;;;;;;;;;;;GAkBG;AACH,MAAM,WAAW,sBAAsB;IACrC,uEAAuE;IACvE,SAAS,EAAE,MAAM,CAAC;IAClB,0DAA0D;IAC1D,QAAQ,EAAE,MAAM,CAAC;IACjB;;;OAGG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB;;;OAGG;IACH,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED;;;;;;;;;;;GAWG;AACH,qBAAa,kBAAkB;IACT,OAAO,CAAC,QAAQ,CAAC,KAAK;IAA1C,OAAO;IAEP;;;;;;OAMG;WACU,MAAM,CAAC,MAAM,EAAE,sBAAsB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAQhF;;;OAGG;IACG,KAAK,IAAI,OAAO,CAAC,MAAM,CAAC;IAI9B;;;;OAIG;IACG,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC;CAGnC"}

package/dist/train.js

@@ -1,48 +0,0 @@
-import { PrivacyLoraTrainerJs as NativePrivacyLoraTrainerJs, } from '@mlx-node/core';
-/**
- * High-level wrapper around the native `PrivacyLoraTrainerJs` NAPI class.
- *
- * The native binding's `create` factory is synchronous (it loads the base
- * checkpoint, attaches zero-B LoRA adapters, and parses the JSONL dataset
- * inline), but the public API here is intentionally `async` so we can later
- * move work to a worker thread without a breaking change to consumers.
- *
- * `train()` and `saveAdapter()` are already async on the native side — they
- * funnel through `spawn_blocking` so the Node.js event loop stays responsive
- * for the duration of the training run.
- */
-export class PrivacyLoraTrainer {
-    inner;
-    constructor(inner) {
-        this.inner = inner;
-    }
-    /**
-     * Construct a new trainer.
-     *
-     * Loads the base privacy-filter model from `config.modelPath`, attaches
-     * zero-B LoRA adapters on every attention projection, and parses the
-     * pre-tokenized JSONL dataset at `config.dataPath`.
-     */
-    static async create(config) {
-        // NAPI requires the camelCase keys to exactly match the interface — pass
-        // the object through directly so any future field additions are picked
-        // up without an extra mapping table.
-        const native = NativePrivacyLoraTrainerJs.create(config);
-        return new PrivacyLoraTrainer(native);
-    }
-    /**
-     * Run the full training loop. Resolves with the final optimizer step
-     * count once `numEpochs` epochs have completed.
-     */
-    async train() {
-        return this.inner.train();
-    }
-    /**
-     * Persist adapter weights, optimizer state, and trainer progress to the
-     * configured `outputDir`. Overwrites any previous checkpoint at that
-     * path.
-     */
-    async saveAdapter() {
-        return this.inner.saveAdapter();
-    }
-}