@huggingface/transformers 4.0.0-next.2 → 4.0.0-next.4

package/dist/transformers.node.mjs

@@ -14,7 +14,7 @@ var __export = (target, all) => {
 import fs from "fs";
 import path from "path";
 import url from "url";
-var VERSION = "4.0.0-next.2";
+var VERSION = "4.0.0-next.4";
 var IS_PROCESS_AVAILABLE = typeof process !== "undefined";
 var IS_NODE_ENV = IS_PROCESS_AVAILABLE && process?.release?.name === "node";
 var IS_FS_AVAILABLE = !isEmpty(fs);
@@ -6759,7 +6759,7 @@ var uint16_to_float32 = /* @__PURE__ */ (function() {
 // src/backends/onnx.js
 import * as ONNX_NODE from "onnxruntime-node";
 
-// ../../node_modules/.pnpm/onnxruntime-web@1.25.0-dev.20260209-a3749f1353/node_modules/onnxruntime-web/dist/ort.webgpu.bundle.min.mjs
+// ../../node_modules/.pnpm/onnxruntime-web@1.25.0-dev.20260212-1a71a5f46e/node_modules/onnxruntime-web/dist/ort.webgpu.bundle.min.mjs
 var ort_webgpu_bundle_min_exports = {};
 __export(ort_webgpu_bundle_min_exports, {
   InferenceSession: () => nc,
@@ -8728,7 +8728,7 @@ async function is(a = {}) {
         we(`invalid type for getValue: ${t}`);
     }
   }, r.UTF8ToString = ct, r.stringToUTF8 = Pe, r.lengthBytesUTF8 = _e;
-  var bo, wo, Rr, $t, Te, mt, go, To, vo, Eo, So, Ao, Io, xo, Lo, Oo, Bo, Nr, Wr, kr, Fr, Et, Gr, Mo, $r, Uo, Co, Do, zr, Po, _o, Vr, k, St, Ro, C, zt, D, No, Hr, Wo, ko, Fo, jr, Go, $o, zo, Vo, Ho, jo, Yo, qo, Jo, Xo, Zo, Ko, Qo, ea, ta, ra, na, oa, aa, sa, ia, ua, fa, ca, da, la, pa, ma, ha, ya, ba, wa, ga, Ta, va, Ea, Sa, Aa, Ia, xa, Ne, df = [Ye, gr, In, Mn, Un, Cn, Dn, Pn, _n, Rn, Nn, Wn, kn, Fn, Gn, $n, eo, to, ro, co, lo, po, mo, ho, yo], Yr = { 1116828: (e, t, n, o, u) => {
+  var bo, wo, Rr, $t, Te, mt, go, To, vo, Eo, So, Ao, Io, xo, Lo, Oo, Bo, Nr, Wr, kr, Fr, Et, Gr, Mo, $r, Uo, Co, Do, zr, Po, _o, Vr, k, St, Ro, C, zt, D, No, Hr, Wo, ko, Fo, jr, Go, $o, zo, Vo, Ho, jo, Yo, qo, Jo, Xo, Zo, Ko, Qo, ea, ta, ra, na, oa, aa, sa, ia, ua, fa, ca, da, la, pa, ma, ha, ya, ba, wa, ga, Ta, va, Ea, Sa, Aa, Ia, xa, Ne, df = [Ye, gr, In, Mn, Un, Cn, Dn, Pn, _n, Rn, Nn, Wn, kn, Fn, Gn, $n, eo, to, ro, co, lo, po, mo, ho, yo], Yr = { 1117404: (e, t, n, o, u) => {
     if (r === void 0 || !r.Zc) return 1;
     if ((e = ct(Number(e >>> 0))).startsWith("./") && (e = e.substring(2)), !(e = r.Zc.get(e))) return 2;
     if (t = Number(t >>> 0), n = Number(n >>> 0), o = Number(o >>> 0), t + n > e.byteLength) return 3;
@@ -8748,11 +8748,11 @@ async function is(a = {}) {
     } catch {
       return 4;
     }
-  }, 1117652: (e, t, n) => {
+  }, 1118228: (e, t, n) => {
     r.ke(e, (l(), J).subarray(t >>> 0, t + n >>> 0));
-  }, 1117716: () => r.Ie(), 1117758: (e) => {
+  }, 1118292: () => r.Ie(), 1118334: (e) => {
     r.je(e);
-  }, 1117795: () => typeof wasmOffsetConverter < "u" };
+  }, 1118371: () => typeof wasmOffsetConverter < "u" };
   function lf() {
     return typeof wasmOffsetConverter < "u";
   }
@@ -10680,7 +10680,7 @@ var qs = F(() => {
 ze();
 ze();
 ze();
-var rs = "1.25.0-dev.20260209-a3749f1353";
+var rs = "1.25.0-dev.20260212-1a71a5f46e";
 var Ol = tn;
 {
   let a = (qs(), Ht(Ys)).wasmBackend;
@@ -19979,6 +19979,7 @@ function getNormalizedConfig(config) {
     case "granite":
     case "granitemoehybrid":
     case "cohere":
+    case "cohere2":
     case "mistral":
     case "starcoder2":
     case "qwen2":
@@ -22314,7 +22315,7 @@ var PreTrainedModel = class extends Callable2 {
   /**
    *
    * @param {GenerationConfig} generation_config
-   * @param {StoppingCriteriaList} [stopping_criteria=null]
+   * @param {import('../generation/stopping_criteria.js').StoppingCriteria|import('../generation/stopping_criteria.js').StoppingCriteria[]|StoppingCriteriaList} [stopping_criteria=null]
    */
   _get_stopping_criteria(generation_config, stopping_criteria = null) {
     const criteria = new StoppingCriteriaList();
@@ -23058,6 +23059,9 @@ __export(models_exports, {
   CodeGenForCausalLM: () => CodeGenForCausalLM,
   CodeGenModel: () => CodeGenModel,
   CodeGenPreTrainedModel: () => CodeGenPreTrainedModel,
+  Cohere2ForCausalLM: () => Cohere2ForCausalLM,
+  Cohere2Model: () => Cohere2Model,
+  Cohere2PreTrainedModel: () => Cohere2PreTrainedModel,
   CohereForCausalLM: () => CohereForCausalLM,
   CohereModel: () => CohereModel,
   CoherePreTrainedModel: () => CoherePreTrainedModel,
@@ -24055,6 +24059,14 @@ var CohereModel = class extends CoherePreTrainedModel {
 var CohereForCausalLM = class extends CoherePreTrainedModel {
 };
 
+// src/models/cohere2/modeling_cohere2.js
+var Cohere2PreTrainedModel = class extends PreTrainedModel {
+};
+var Cohere2Model = class extends Cohere2PreTrainedModel {
+};
+var Cohere2ForCausalLM = class extends Cohere2PreTrainedModel {
+};
+
 // src/models/convbert/modeling_convbert.js
 var ConvBertPreTrainedModel = class extends PreTrainedModel {
 };
@@ -27859,6 +27871,7 @@ var MODEL_MAPPING_NAMES_DECODER_ONLY = /* @__PURE__ */ new Map([
   ["granite", "GraniteModel"],
   ["granitemoehybrid", "GraniteMoeHybridModel"],
   ["cohere", "CohereModel"],
+  ["cohere2", "Cohere2Model"],
   ["gemma", "GemmaModel"],
   ["gemma2", "Gemma2Model"],
   ["vaultgemma", "VaultGemmaModel"],
@@ -27972,6 +27985,7 @@ var MODEL_FOR_CAUSAL_LM_MAPPING_NAMES = /* @__PURE__ */ new Map([
   ["granite", "GraniteForCausalLM"],
   ["granitemoehybrid", "GraniteMoeHybridForCausalLM"],
   ["cohere", "CohereForCausalLM"],
+  ["cohere2", "Cohere2ForCausalLM"],
   ["gemma", "GemmaForCausalLM"],
   ["gemma2", "Gemma2ForCausalLM"],
   ["vaultgemma", "VaultGemmaForCausalLM"],
@@ -30123,6 +30137,7 @@ var TextStreamer = class extends BaseStreamer {
     this.token_cache = [];
     this.print_len = 0;
     this.next_tokens_are_prompt = true;
+    this.special_ids = new Set(this.tokenizer.all_special_ids.map(BigInt));
   }
   /**
    * Receives tokens, decodes them, and prints them to stdout as soon as they form entire words.
@@ -30139,6 +30154,19 @@ var TextStreamer = class extends BaseStreamer {
     }
     const tokens = value[0];
     this.token_callback_function?.(tokens);
+    if (tokens.length === 1 && this.special_ids.has(tokens[0])) {
+      if (this.decode_kwargs.skip_special_tokens) return;
+      if (this.token_cache.length > 0) {
+        const text2 = this.tokenizer.decode(this.token_cache, this.decode_kwargs);
+        const printable_text2 = text2.slice(this.print_len);
+        this.on_finalized_text(printable_text2, false);
+        this.token_cache = [];
+        this.print_len = 0;
+      }
+      const special_text = this.tokenizer.decode(tokens, this.decode_kwargs);
+      this.on_finalized_text(special_text, false);
+      return;
+    }
     this.token_cache = mergeArrays(this.token_cache, tokens);
     const text = this.tokenizer.decode(this.token_cache, this.decode_kwargs);
     let printable_text;
@@ -30469,6 +30497,9 @@ export {
   CodeGenPreTrainedModel,
   CodeGenTokenizer,
   CodeLlamaTokenizer,
+  Cohere2ForCausalLM,
+  Cohere2Model,
+  Cohere2PreTrainedModel,
   CohereForCausalLM,
   CohereModel,
   CoherePreTrainedModel,
@@ -31197,7 +31228,7 @@ export {
 
 onnxruntime-web/dist/ort.webgpu.bundle.min.mjs:
   (*!
-   * ONNX Runtime Web v1.25.0-dev.20260209-a3749f1353
+   * ONNX Runtime Web v1.25.0-dev.20260212-1a71a5f46e
    * Copyright (c) Microsoft Corporation. All rights reserved.
    * Licensed under the MIT License.
    *)

package/dist/transformers.web.js

@@ -17,7 +17,7 @@ var emptyObj3 = {};
 var node_url_default = emptyObj3;
 
 // src/env.js
-var VERSION = "4.0.0-next.2";
+var VERSION = "4.0.0-next.4";
 var IS_PROCESS_AVAILABLE = typeof process !== "undefined";
 var IS_NODE_ENV = IS_PROCESS_AVAILABLE && process?.release?.name === "node";
 var IS_FS_AVAILABLE = !isEmpty(node_fs_default);
@@ -16079,6 +16079,7 @@ function getNormalizedConfig(config) {
     case "granite":
     case "granitemoehybrid":
     case "cohere":
+    case "cohere2":
     case "mistral":
     case "starcoder2":
     case "qwen2":
@@ -18414,7 +18415,7 @@ var PreTrainedModel = class extends Callable2 {
   /**
    *
    * @param {GenerationConfig} generation_config
-   * @param {StoppingCriteriaList} [stopping_criteria=null]
+   * @param {import('../generation/stopping_criteria.js').StoppingCriteria|import('../generation/stopping_criteria.js').StoppingCriteria[]|StoppingCriteriaList} [stopping_criteria=null]
    */
   _get_stopping_criteria(generation_config, stopping_criteria = null) {
     const criteria = new StoppingCriteriaList();
@@ -19158,6 +19159,9 @@ __export(models_exports, {
   CodeGenForCausalLM: () => CodeGenForCausalLM,
   CodeGenModel: () => CodeGenModel,
   CodeGenPreTrainedModel: () => CodeGenPreTrainedModel,
+  Cohere2ForCausalLM: () => Cohere2ForCausalLM,
+  Cohere2Model: () => Cohere2Model,
+  Cohere2PreTrainedModel: () => Cohere2PreTrainedModel,
   CohereForCausalLM: () => CohereForCausalLM,
   CohereModel: () => CohereModel,
   CoherePreTrainedModel: () => CoherePreTrainedModel,
@@ -20155,6 +20159,14 @@ var CohereModel = class extends CoherePreTrainedModel {
 var CohereForCausalLM = class extends CoherePreTrainedModel {
 };
 
+// src/models/cohere2/modeling_cohere2.js
+var Cohere2PreTrainedModel = class extends PreTrainedModel {
+};
+var Cohere2Model = class extends Cohere2PreTrainedModel {
+};
+var Cohere2ForCausalLM = class extends Cohere2PreTrainedModel {
+};
+
 // src/models/convbert/modeling_convbert.js
 var ConvBertPreTrainedModel = class extends PreTrainedModel {
 };
@@ -23959,6 +23971,7 @@ var MODEL_MAPPING_NAMES_DECODER_ONLY = /* @__PURE__ */ new Map([
   ["granite", "GraniteModel"],
   ["granitemoehybrid", "GraniteMoeHybridModel"],
   ["cohere", "CohereModel"],
+  ["cohere2", "Cohere2Model"],
   ["gemma", "GemmaModel"],
   ["gemma2", "Gemma2Model"],
   ["vaultgemma", "VaultGemmaModel"],
@@ -24072,6 +24085,7 @@ var MODEL_FOR_CAUSAL_LM_MAPPING_NAMES = /* @__PURE__ */ new Map([
   ["granite", "GraniteForCausalLM"],
   ["granitemoehybrid", "GraniteMoeHybridForCausalLM"],
   ["cohere", "CohereForCausalLM"],
+  ["cohere2", "Cohere2ForCausalLM"],
   ["gemma", "GemmaForCausalLM"],
   ["gemma2", "Gemma2ForCausalLM"],
   ["vaultgemma", "VaultGemmaForCausalLM"],
@@ -26223,6 +26237,7 @@ var TextStreamer = class extends BaseStreamer {
     this.token_cache = [];
     this.print_len = 0;
     this.next_tokens_are_prompt = true;
+    this.special_ids = new Set(this.tokenizer.all_special_ids.map(BigInt));
   }
   /**
    * Receives tokens, decodes them, and prints them to stdout as soon as they form entire words.
@@ -26239,6 +26254,19 @@ var TextStreamer = class extends BaseStreamer {
     }
     const tokens = value[0];
     this.token_callback_function?.(tokens);
+    if (tokens.length === 1 && this.special_ids.has(tokens[0])) {
+      if (this.decode_kwargs.skip_special_tokens) return;
+      if (this.token_cache.length > 0) {
+        const text2 = this.tokenizer.decode(this.token_cache, this.decode_kwargs);
+        const printable_text2 = text2.slice(this.print_len);
+        this.on_finalized_text(printable_text2, false);
+        this.token_cache = [];
+        this.print_len = 0;
+      }
+      const special_text = this.tokenizer.decode(tokens, this.decode_kwargs);
+      this.on_finalized_text(special_text, false);
+      return;
+    }
     this.token_cache = mergeArrays(this.token_cache, tokens);
     const text = this.tokenizer.decode(this.token_cache, this.decode_kwargs);
     let printable_text;
@@ -26569,6 +26597,9 @@ export {
   CodeGenPreTrainedModel,
   CodeGenTokenizer,
   CodeLlamaTokenizer,
+  Cohere2ForCausalLM,
+  Cohere2Model,
+  Cohere2PreTrainedModel,
   CohereForCausalLM,
   CohereModel,
   CoherePreTrainedModel,