@huggingface/tasks 0.12.27 → 0.12.29

package/dist/index.cjs

@@ -24,6 +24,9 @@ __export(src_exports, {
   ALL_MODEL_LIBRARY_KEYS: () => ALL_MODEL_LIBRARY_KEYS,
   DATASET_LIBRARIES_UI_ELEMENTS: () => DATASET_LIBRARIES_UI_ELEMENTS,
   DEFAULT_MEMORY_OPTIONS: () => DEFAULT_MEMORY_OPTIONS,
+  GGMLQuantizationType: () => GGMLQuantizationType,
+  GGUF_QUANT_RE: () => GGUF_QUANT_RE,
+  GGUF_QUANT_RE_GLOBAL: () => GGUF_QUANT_RE_GLOBAL,
   LIBRARY_TASK_MAPPING: () => LIBRARY_TASK_MAPPING,
   LOCAL_APPS: () => LOCAL_APPS,
   MAPPING_DEFAULT_WIDGET: () => MAPPING_DEFAULT_WIDGET,
@@ -38,6 +41,7 @@ __export(src_exports, {
   SUBTASK_TYPES: () => SUBTASK_TYPES,
   TASKS_DATA: () => TASKS_DATA,
   TASKS_MODEL_LIBRARIES: () => TASKS_MODEL_LIBRARIES,
+  parseGGUFQuantLabel: () => parseGGUFQuantLabel,
   snippets: () => snippets_exports
 });
 module.exports = __toCommonJS(src_exports);
@@ -795,7 +799,7 @@ var MAPPING_DEFAULT_WIDGET = /* @__PURE__ */ new Map([
 ]);
 
 // src/pipelines.ts
-var MODALITIES = ["cv", "nlp", "audio", "tabular", "multimodal", "rl", "other"];
+var MODALITIES = ["multimodal", "nlp", "cv", "audio", "tabular", "rl", "other"];
 var MODALITY_LABELS = {
   multimodal: "Multimodal",
   nlp: "Natural Language Processing",
@@ -4835,9 +4839,9 @@ var gliner = (model) => [
 model = GLiNER.from_pretrained("${model.id}")`
 ];
 var keras = (model) => [
-  `# Available backend options are: "jax", "tensorflow", "torch".
+  `# Available backend options are: "jax", "torch", "tensorflow".
 import os
-os.environ["KERAS_BACKEND"] = "tensorflow"
+os.environ["KERAS_BACKEND"] = "jax"
 	
 import keras
 
@@ -4845,9 +4849,9 @@ model = keras.saving.load_model("hf://${model.id}")
 `
 ];
 var keras_nlp = (model) => [
-  `# Available backend options are: "jax", "tensorflow", "torch".
+  `# Available backend options are: "jax", "torch", "tensorflow".
 import os
-os.environ["KERAS_BACKEND"] = "tensorflow"
+os.environ["KERAS_BACKEND"] = "jax"
 
 import keras_nlp
 
@@ -4855,6 +4859,20 @@ tokenizer = keras_nlp.models.Tokenizer.from_preset("hf://${model.id}")
 backbone = keras_nlp.models.Backbone.from_preset("hf://${model.id}")
 `
 ];
+var keras_hub = (model) => [
+  `# Available backend options are: "jax", "torch", "tensorflow".
+import os
+os.environ["KERAS_BACKEND"] = "jax"
+
+import keras_hub
+
+# Load a task-specific model (*replace CausalLM with your task*)
+model = keras_hub.models.CausalLM.from_preset("hf://${model.id}", dtype="bfloat16")
+
+# Possible tasks are CausalLM, TextToImage, ImageClassifier, ...
+# full list here: https://keras.io/api/keras_hub/models/#api-documentation
+`
+];
 var llama_cpp_python = (model) => [
   `from llama_cpp import Llama
 
@@ -5769,16 +5787,23 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     repoUrl: "https://github.com/keras-team/tf-keras",
     docsUrl: "https://huggingface.co/docs/hub/tf-keras",
     snippets: tf_keras,
-    filter: true,
     countDownloads: `path:"saved_model.pb"`
   },
   "keras-nlp": {
     prettyLabel: "KerasNLP",
     repoName: "KerasNLP",
-    repoUrl: "https://keras.io/keras_nlp/",
-    docsUrl: "https://github.com/keras-team/keras-nlp",
+    repoUrl: "https://github.com/keras-team/keras-nlp",
+    docsUrl: "https://keras.io/keras_nlp/",
     snippets: keras_nlp
   },
+  "keras-hub": {
+    prettyLabel: "KerasHub",
+    repoName: "KerasHub",
+    repoUrl: "https://github.com/keras-team/keras-hub",
+    docsUrl: "https://keras.io/keras_hub/",
+    snippets: keras_hub,
+    filter: true
+  },
   k2: {
     prettyLabel: "K2",
     repoName: "k2",
@@ -5797,6 +5822,12 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     repoUrl: "https://github.com/abetlen/llama-cpp-python",
     snippets: llama_cpp_python
   },
+  "mini-omni2": {
+    prettyLabel: "Mini-Omni2",
+    repoName: "Mini-Omni2",
+    repoUrl: "https://github.com/gpt-omni/mini-omni2",
+    countDownloads: `path:"model_config.yaml"`
+  },
   mindspore: {
     prettyLabel: "MindSpore",
     repoName: "mindspore",
@@ -5880,6 +5911,12 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     filter: true,
     countDownloads: `path_extension:"nemo" OR path:"model_config.yaml"`
   },
+  "open-oasis": {
+    prettyLabel: "open-oasis",
+    repoName: "open-oasis",
+    repoUrl: "https://github.com/etched-ai/open-oasis",
+    countDownloads: `path:"oasis500m.pt"`
+  },
   open_clip: {
     prettyLabel: "OpenCLIP",
     repoName: "OpenCLIP",
@@ -6104,6 +6141,13 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     filter: false,
     countDownloads: `path_extension:"safetensors" OR path_extension:"pt"`
   },
+  genmo: {
+    prettyLabel: "Genmo",
+    repoName: "Genmo",
+    repoUrl: "https://github.com/genmoai/models",
+    filter: false,
+    countDownloads: `path:"vae_stats.json"`
+  },
   tensorflowtts: {
     prettyLabel: "TensorFlowTTS",
     repoName: "TensorFlowTTS",
@@ -6250,7 +6294,28 @@ var inputsQuestionAnswering = () => `{
 }`;
 var inputsTextClassification = () => `"I like you. I love you"`;
 var inputsTokenClassification = () => `"My name is Sarah Jessica Parker but you can call me Jessica"`;
-var inputsTextGeneration = () => `"Can you please let us know more details about your "`;
+var inputsTextGeneration = (model) => {
+  if (model.tags.includes("conversational")) {
+    return model.pipeline_tag === "text-generation" ? [{ role: "user", content: "What is the capital of France?" }] : [
+      {
+        role: "user",
+        content: [
+          {
+            type: "text",
+            text: "Describe this image in one sentence."
+          },
+          {
+            type: "image_url",
+            image_url: {
+              url: "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg"
+            }
+          }
+        ]
+      }
+    ];
+  }
+  return `"Can you please let us know more details about your "`;
+};
 var inputsText2TextGeneration = () => `"The answer to the universe is"`;
 var inputsFillMask = (model) => `"The answer to the universe is ${model.mask_token}."`;
 var inputsSentenceSimilarity = () => `{
@@ -6307,13 +6372,15 @@ function getModelInputSnippet(model, noWrap = false, noQuotes = false) {
     const inputs = modelInputSnippets[model.pipeline_tag];
     if (inputs) {
       let result = inputs(model);
-      if (noWrap) {
-        result = result.replace(/(?:(?:\r?\n|\r)\t*)|\t+/g, " ");
-      }
-      if (noQuotes) {
-        const REGEX_QUOTES = /^"(.+)"$/s;
-        const match = result.match(REGEX_QUOTES);
-        result = match ? match[1] : result;
+      if (typeof result === "string") {
+        if (noWrap) {
+          result = result.replace(/(?:(?:\r?\n|\r)\t*)|\t+/g, " ");
+        }
+        if (noQuotes) {
+          const REGEX_QUOTES = /^"(.+)"$/s;
+          const match = result.match(REGEX_QUOTES);
+          result = match ? match[1] : result;
+        }
       }
       return result;
     }
@@ -6355,7 +6422,7 @@ function stringifyMessages(messages, opts) {
       if (opts.customContentEscaper) {
         content = opts.customContentEscaper(content);
       }
-      return `{ ${keyRole}: "${role}", ${keyContent}: ${content} }`;
+      return `{ ${keyRole}: "${role}", ${keyContent}: [${content}] }`;
     }
   });
   return opts.start + messagesStringified.join(opts.sep) + opts.end;
@@ -6376,20 +6443,7 @@ var snippetBasic = (model, accessToken) => ({
 var snippetTextGeneration = (model, accessToken, opts) => {
   if (model.tags.includes("conversational")) {
     const streaming = opts?.streaming ?? true;
-    const exampleMessages = model.pipeline_tag === "text-generation" ? [{ role: "user", content: "What is the capital of France?" }] : [
-      {
-        role: "user",
-        content: [
-          {
-            type: "image_url",
-            image_url: {
-              url: "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg"
-            }
-          },
-          { type: "text", text: "Describe this image in one sentence." }
-        ]
-      }
-    ];
+    const exampleMessages = getModelInputSnippet(model);
     const messages = opts?.messages ?? exampleMessages;
     const config = {
       ...opts?.temperature ? { temperature: opts.temperature } : void 0,
@@ -6489,20 +6543,7 @@ __export(python_exports, {
 });
 var snippetConversational = (model, accessToken, opts) => {
   const streaming = opts?.streaming ?? true;
-  const exampleMessages = model.pipeline_tag === "text-generation" ? [{ role: "user", content: "What is the capital of France?" }] : [
-    {
-      role: "user",
-      content: [
-        {
-          type: "image_url",
-          image_url: {
-            url: "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg"
-          }
-        },
-        { type: "text", text: "Describe this image in one sentence." }
-      ]
-    }
-  ];
+  const exampleMessages = getModelInputSnippet(model);
   const messages = opts?.messages ?? exampleMessages;
   const messagesStr = stringifyMessages(messages, {
     sep: ",\n	",
@@ -6800,20 +6841,7 @@ query({"inputs": ${getModelInputSnippet(model)}}).then((response) => {
 var snippetTextGeneration2 = (model, accessToken, opts) => {
   if (model.tags.includes("conversational")) {
     const streaming = opts?.streaming ?? true;
-    const exampleMessages = model.pipeline_tag === "text-generation" ? [{ role: "user", content: "What is the capital of France?" }] : [
-      {
-        role: "user",
-        content: [
-          {
-            type: "image_url",
-            image_url: {
-              url: "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg"
-            }
-          },
-          { type: "text", text: "Describe this image in one sentence." }
-        ]
-      }
-    ];
+    const exampleMessages = getModelInputSnippet(model);
     const messages = opts?.messages ?? exampleMessages;
     const messagesStr = stringifyMessages(messages, { sep: ",\n		", start: "[\n		", end: "\n	]" });
     const config = {
@@ -7051,6 +7079,47 @@ function hasJsInferenceSnippet(model) {
   return !!model.pipeline_tag && model.pipeline_tag in jsSnippets;
 }
 
+// src/gguf.ts
+var GGMLQuantizationType = /* @__PURE__ */ ((GGMLQuantizationType2) => {
+  GGMLQuantizationType2[GGMLQuantizationType2["F32"] = 0] = "F32";
+  GGMLQuantizationType2[GGMLQuantizationType2["F16"] = 1] = "F16";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q4_0"] = 2] = "Q4_0";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q4_1"] = 3] = "Q4_1";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q5_0"] = 6] = "Q5_0";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q5_1"] = 7] = "Q5_1";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q8_0"] = 8] = "Q8_0";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q8_1"] = 9] = "Q8_1";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q2_K"] = 10] = "Q2_K";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q3_K"] = 11] = "Q3_K";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q4_K"] = 12] = "Q4_K";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q5_K"] = 13] = "Q5_K";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q6_K"] = 14] = "Q6_K";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q8_K"] = 15] = "Q8_K";
+  GGMLQuantizationType2[GGMLQuantizationType2["IQ2_XXS"] = 16] = "IQ2_XXS";
+  GGMLQuantizationType2[GGMLQuantizationType2["IQ2_XS"] = 17] = "IQ2_XS";
+  GGMLQuantizationType2[GGMLQuantizationType2["IQ3_XXS"] = 18] = "IQ3_XXS";
+  GGMLQuantizationType2[GGMLQuantizationType2["IQ1_S"] = 19] = "IQ1_S";
+  GGMLQuantizationType2[GGMLQuantizationType2["IQ4_NL"] = 20] = "IQ4_NL";
+  GGMLQuantizationType2[GGMLQuantizationType2["IQ3_S"] = 21] = "IQ3_S";
+  GGMLQuantizationType2[GGMLQuantizationType2["IQ2_S"] = 22] = "IQ2_S";
+  GGMLQuantizationType2[GGMLQuantizationType2["IQ4_XS"] = 23] = "IQ4_XS";
+  GGMLQuantizationType2[GGMLQuantizationType2["I8"] = 24] = "I8";
+  GGMLQuantizationType2[GGMLQuantizationType2["I16"] = 25] = "I16";
+  GGMLQuantizationType2[GGMLQuantizationType2["I32"] = 26] = "I32";
+  GGMLQuantizationType2[GGMLQuantizationType2["I64"] = 27] = "I64";
+  GGMLQuantizationType2[GGMLQuantizationType2["F64"] = 28] = "F64";
+  GGMLQuantizationType2[GGMLQuantizationType2["IQ1_M"] = 29] = "IQ1_M";
+  GGMLQuantizationType2[GGMLQuantizationType2["BF16"] = 30] = "BF16";
+  return GGMLQuantizationType2;
+})(GGMLQuantizationType || {});
+var ggufQuants = Object.values(GGMLQuantizationType).filter((v) => typeof v === "string");
+var GGUF_QUANT_RE = new RegExp(`(?<quant>${ggufQuants.join("|")})(_(?<sizeVariation>[A-Z]+))?`);
+var GGUF_QUANT_RE_GLOBAL = new RegExp(GGUF_QUANT_RE, "g");
+function parseGGUFQuantLabel(fname) {
+  const quantLabel = fname.toUpperCase().match(GGUF_QUANT_RE_GLOBAL)?.at(-1);
+  return quantLabel;
+}
+
 // src/hardware.ts
 var TFLOPS_THRESHOLD_WHITE_HOUSE_MODEL_TRAINING_TOTAL = 10 ** 14;
 var TFLOPS_THRESHOLD_WHITE_HOUSE_MODEL_TRAINING_TOTAL_BIOLOGY = 10 ** 11;
@@ -7509,7 +7578,6 @@ var SKUS = {
 };
 
 // src/local-apps.ts
-var import_gguf = require("@huggingface/gguf");
 function isAwqModel(model) {
   return model.config?.quantization_config?.quant_method === "awq";
 }
@@ -7587,7 +7655,7 @@ var snippetNodeLlamaCppCli = (model, filepath) => {
 };
 var snippetOllama = (model, filepath) => {
   if (filepath) {
-    const quantLabel = (0, import_gguf.parseGGUFQuantLabel)(filepath);
+    const quantLabel = parseGGUFQuantLabel(filepath);
     const ollamatag = quantLabel ? `:${quantLabel}` : "";
     return `ollama run hf.co/${model.id}${ollamatag}`;
   }
@@ -7891,6 +7959,9 @@ var DATASET_LIBRARIES_UI_ELEMENTS = {
   ALL_MODEL_LIBRARY_KEYS,
   DATASET_LIBRARIES_UI_ELEMENTS,
   DEFAULT_MEMORY_OPTIONS,
+  GGMLQuantizationType,
+  GGUF_QUANT_RE,
+  GGUF_QUANT_RE_GLOBAL,
   LIBRARY_TASK_MAPPING,
   LOCAL_APPS,
   MAPPING_DEFAULT_WIDGET,
@@ -7905,5 +7976,6 @@ var DATASET_LIBRARIES_UI_ELEMENTS = {
   SUBTASK_TYPES,
   TASKS_DATA,
   TASKS_MODEL_LIBRARIES,
+  parseGGUFQuantLabel,
   snippets
 });

package/dist/index.js

@@ -757,7 +757,7 @@ var MAPPING_DEFAULT_WIDGET = /* @__PURE__ */ new Map([
 ]);
 
 // src/pipelines.ts
-var MODALITIES = ["cv", "nlp", "audio", "tabular", "multimodal", "rl", "other"];
+var MODALITIES = ["multimodal", "nlp", "cv", "audio", "tabular", "rl", "other"];
 var MODALITY_LABELS = {
   multimodal: "Multimodal",
   nlp: "Natural Language Processing",
@@ -4797,9 +4797,9 @@ var gliner = (model) => [
 model = GLiNER.from_pretrained("${model.id}")`
 ];
 var keras = (model) => [
-  `# Available backend options are: "jax", "tensorflow", "torch".
+  `# Available backend options are: "jax", "torch", "tensorflow".
 import os
-os.environ["KERAS_BACKEND"] = "tensorflow"
+os.environ["KERAS_BACKEND"] = "jax"
 	
 import keras
 
@@ -4807,9 +4807,9 @@ model = keras.saving.load_model("hf://${model.id}")
 `
 ];
 var keras_nlp = (model) => [
-  `# Available backend options are: "jax", "tensorflow", "torch".
+  `# Available backend options are: "jax", "torch", "tensorflow".
 import os
-os.environ["KERAS_BACKEND"] = "tensorflow"
+os.environ["KERAS_BACKEND"] = "jax"
 
 import keras_nlp
 
@@ -4817,6 +4817,20 @@ tokenizer = keras_nlp.models.Tokenizer.from_preset("hf://${model.id}")
 backbone = keras_nlp.models.Backbone.from_preset("hf://${model.id}")
 `
 ];
+var keras_hub = (model) => [
+  `# Available backend options are: "jax", "torch", "tensorflow".
+import os
+os.environ["KERAS_BACKEND"] = "jax"
+
+import keras_hub
+
+# Load a task-specific model (*replace CausalLM with your task*)
+model = keras_hub.models.CausalLM.from_preset("hf://${model.id}", dtype="bfloat16")
+
+# Possible tasks are CausalLM, TextToImage, ImageClassifier, ...
+# full list here: https://keras.io/api/keras_hub/models/#api-documentation
+`
+];
 var llama_cpp_python = (model) => [
   `from llama_cpp import Llama
 
@@ -5731,16 +5745,23 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     repoUrl: "https://github.com/keras-team/tf-keras",
     docsUrl: "https://huggingface.co/docs/hub/tf-keras",
     snippets: tf_keras,
-    filter: true,
     countDownloads: `path:"saved_model.pb"`
   },
   "keras-nlp": {
     prettyLabel: "KerasNLP",
     repoName: "KerasNLP",
-    repoUrl: "https://keras.io/keras_nlp/",
-    docsUrl: "https://github.com/keras-team/keras-nlp",
+    repoUrl: "https://github.com/keras-team/keras-nlp",
+    docsUrl: "https://keras.io/keras_nlp/",
     snippets: keras_nlp
   },
+  "keras-hub": {
+    prettyLabel: "KerasHub",
+    repoName: "KerasHub",
+    repoUrl: "https://github.com/keras-team/keras-hub",
+    docsUrl: "https://keras.io/keras_hub/",
+    snippets: keras_hub,
+    filter: true
+  },
   k2: {
     prettyLabel: "K2",
     repoName: "k2",
@@ -5759,6 +5780,12 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     repoUrl: "https://github.com/abetlen/llama-cpp-python",
     snippets: llama_cpp_python
   },
+  "mini-omni2": {
+    prettyLabel: "Mini-Omni2",
+    repoName: "Mini-Omni2",
+    repoUrl: "https://github.com/gpt-omni/mini-omni2",
+    countDownloads: `path:"model_config.yaml"`
+  },
   mindspore: {
     prettyLabel: "MindSpore",
     repoName: "mindspore",
@@ -5842,6 +5869,12 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     filter: true,
     countDownloads: `path_extension:"nemo" OR path:"model_config.yaml"`
   },
+  "open-oasis": {
+    prettyLabel: "open-oasis",
+    repoName: "open-oasis",
+    repoUrl: "https://github.com/etched-ai/open-oasis",
+    countDownloads: `path:"oasis500m.pt"`
+  },
   open_clip: {
     prettyLabel: "OpenCLIP",
     repoName: "OpenCLIP",
@@ -6066,6 +6099,13 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     filter: false,
     countDownloads: `path_extension:"safetensors" OR path_extension:"pt"`
   },
+  genmo: {
+    prettyLabel: "Genmo",
+    repoName: "Genmo",
+    repoUrl: "https://github.com/genmoai/models",
+    filter: false,
+    countDownloads: `path:"vae_stats.json"`
+  },
   tensorflowtts: {
     prettyLabel: "TensorFlowTTS",
     repoName: "TensorFlowTTS",
@@ -6212,7 +6252,28 @@ var inputsQuestionAnswering = () => `{
 }`;
 var inputsTextClassification = () => `"I like you. I love you"`;
 var inputsTokenClassification = () => `"My name is Sarah Jessica Parker but you can call me Jessica"`;
-var inputsTextGeneration = () => `"Can you please let us know more details about your "`;
+var inputsTextGeneration = (model) => {
+  if (model.tags.includes("conversational")) {
+    return model.pipeline_tag === "text-generation" ? [{ role: "user", content: "What is the capital of France?" }] : [
+      {
+        role: "user",
+        content: [
+          {
+            type: "text",
+            text: "Describe this image in one sentence."
+          },
+          {
+            type: "image_url",
+            image_url: {
+              url: "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg"
+            }
+          }
+        ]
+      }
+    ];
+  }
+  return `"Can you please let us know more details about your "`;
+};
 var inputsText2TextGeneration = () => `"The answer to the universe is"`;
 var inputsFillMask = (model) => `"The answer to the universe is ${model.mask_token}."`;
 var inputsSentenceSimilarity = () => `{
@@ -6269,13 +6330,15 @@ function getModelInputSnippet(model, noWrap = false, noQuotes = false) {
     const inputs = modelInputSnippets[model.pipeline_tag];
     if (inputs) {
       let result = inputs(model);
-      if (noWrap) {
-        result = result.replace(/(?:(?:\r?\n|\r)\t*)|\t+/g, " ");
-      }
-      if (noQuotes) {
-        const REGEX_QUOTES = /^"(.+)"$/s;
-        const match = result.match(REGEX_QUOTES);
-        result = match ? match[1] : result;
+      if (typeof result === "string") {
+        if (noWrap) {
+          result = result.replace(/(?:(?:\r?\n|\r)\t*)|\t+/g, " ");
+        }
+        if (noQuotes) {
+          const REGEX_QUOTES = /^"(.+)"$/s;
+          const match = result.match(REGEX_QUOTES);
+          result = match ? match[1] : result;
+        }
       }
       return result;
     }
@@ -6317,7 +6380,7 @@ function stringifyMessages(messages, opts) {
       if (opts.customContentEscaper) {
         content = opts.customContentEscaper(content);
       }
-      return `{ ${keyRole}: "${role}", ${keyContent}: ${content} }`;
+      return `{ ${keyRole}: "${role}", ${keyContent}: [${content}] }`;
     }
   });
   return opts.start + messagesStringified.join(opts.sep) + opts.end;
@@ -6338,20 +6401,7 @@ var snippetBasic = (model, accessToken) => ({
 var snippetTextGeneration = (model, accessToken, opts) => {
   if (model.tags.includes("conversational")) {
     const streaming = opts?.streaming ?? true;
-    const exampleMessages = model.pipeline_tag === "text-generation" ? [{ role: "user", content: "What is the capital of France?" }] : [
-      {
-        role: "user",
-        content: [
-          {
-            type: "image_url",
-            image_url: {
-              url: "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg"
-            }
-          },
-          { type: "text", text: "Describe this image in one sentence." }
-        ]
-      }
-    ];
+    const exampleMessages = getModelInputSnippet(model);
     const messages = opts?.messages ?? exampleMessages;
     const config = {
       ...opts?.temperature ? { temperature: opts.temperature } : void 0,
@@ -6451,20 +6501,7 @@ __export(python_exports, {
 });
 var snippetConversational = (model, accessToken, opts) => {
   const streaming = opts?.streaming ?? true;
-  const exampleMessages = model.pipeline_tag === "text-generation" ? [{ role: "user", content: "What is the capital of France?" }] : [
-    {
-      role: "user",
-      content: [
-        {
-          type: "image_url",
-          image_url: {
-            url: "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg"
-          }
-        },
-        { type: "text", text: "Describe this image in one sentence." }
-      ]
-    }
-  ];
+  const exampleMessages = getModelInputSnippet(model);
   const messages = opts?.messages ?? exampleMessages;
   const messagesStr = stringifyMessages(messages, {
     sep: ",\n	",
@@ -6762,20 +6799,7 @@ query({"inputs": ${getModelInputSnippet(model)}}).then((response) => {
 var snippetTextGeneration2 = (model, accessToken, opts) => {
   if (model.tags.includes("conversational")) {
     const streaming = opts?.streaming ?? true;
-    const exampleMessages = model.pipeline_tag === "text-generation" ? [{ role: "user", content: "What is the capital of France?" }] : [
-      {
-        role: "user",
-        content: [
-          {
-            type: "image_url",
-            image_url: {
-              url: "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg"
-            }
-          },
-          { type: "text", text: "Describe this image in one sentence." }
-        ]
-      }
-    ];
+    const exampleMessages = getModelInputSnippet(model);
     const messages = opts?.messages ?? exampleMessages;
     const messagesStr = stringifyMessages(messages, { sep: ",\n		", start: "[\n		", end: "\n	]" });
     const config = {
@@ -7013,6 +7037,47 @@ function hasJsInferenceSnippet(model) {
   return !!model.pipeline_tag && model.pipeline_tag in jsSnippets;
 }
 
+// src/gguf.ts
+var GGMLQuantizationType = /* @__PURE__ */ ((GGMLQuantizationType2) => {
+  GGMLQuantizationType2[GGMLQuantizationType2["F32"] = 0] = "F32";
+  GGMLQuantizationType2[GGMLQuantizationType2["F16"] = 1] = "F16";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q4_0"] = 2] = "Q4_0";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q4_1"] = 3] = "Q4_1";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q5_0"] = 6] = "Q5_0";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q5_1"] = 7] = "Q5_1";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q8_0"] = 8] = "Q8_0";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q8_1"] = 9] = "Q8_1";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q2_K"] = 10] = "Q2_K";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q3_K"] = 11] = "Q3_K";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q4_K"] = 12] = "Q4_K";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q5_K"] = 13] = "Q5_K";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q6_K"] = 14] = "Q6_K";
+  GGMLQuantizationType2[GGMLQuantizationType2["Q8_K"] = 15] = "Q8_K";
+  GGMLQuantizationType2[GGMLQuantizationType2["IQ2_XXS"] = 16] = "IQ2_XXS";
+  GGMLQuantizationType2[GGMLQuantizationType2["IQ2_XS"] = 17] = "IQ2_XS";
+  GGMLQuantizationType2[GGMLQuantizationType2["IQ3_XXS"] = 18] = "IQ3_XXS";
+  GGMLQuantizationType2[GGMLQuantizationType2["IQ1_S"] = 19] = "IQ1_S";
+  GGMLQuantizationType2[GGMLQuantizationType2["IQ4_NL"] = 20] = "IQ4_NL";
+  GGMLQuantizationType2[GGMLQuantizationType2["IQ3_S"] = 21] = "IQ3_S";
+  GGMLQuantizationType2[GGMLQuantizationType2["IQ2_S"] = 22] = "IQ2_S";
+  GGMLQuantizationType2[GGMLQuantizationType2["IQ4_XS"] = 23] = "IQ4_XS";
+  GGMLQuantizationType2[GGMLQuantizationType2["I8"] = 24] = "I8";
+  GGMLQuantizationType2[GGMLQuantizationType2["I16"] = 25] = "I16";
+  GGMLQuantizationType2[GGMLQuantizationType2["I32"] = 26] = "I32";
+  GGMLQuantizationType2[GGMLQuantizationType2["I64"] = 27] = "I64";
+  GGMLQuantizationType2[GGMLQuantizationType2["F64"] = 28] = "F64";
+  GGMLQuantizationType2[GGMLQuantizationType2["IQ1_M"] = 29] = "IQ1_M";
+  GGMLQuantizationType2[GGMLQuantizationType2["BF16"] = 30] = "BF16";
+  return GGMLQuantizationType2;
+})(GGMLQuantizationType || {});
+var ggufQuants = Object.values(GGMLQuantizationType).filter((v) => typeof v === "string");
+var GGUF_QUANT_RE = new RegExp(`(?<quant>${ggufQuants.join("|")})(_(?<sizeVariation>[A-Z]+))?`);
+var GGUF_QUANT_RE_GLOBAL = new RegExp(GGUF_QUANT_RE, "g");
+function parseGGUFQuantLabel(fname) {
+  const quantLabel = fname.toUpperCase().match(GGUF_QUANT_RE_GLOBAL)?.at(-1);
+  return quantLabel;
+}
+
 // src/hardware.ts
 var TFLOPS_THRESHOLD_WHITE_HOUSE_MODEL_TRAINING_TOTAL = 10 ** 14;
 var TFLOPS_THRESHOLD_WHITE_HOUSE_MODEL_TRAINING_TOTAL_BIOLOGY = 10 ** 11;
@@ -7471,7 +7536,6 @@ var SKUS = {
 };
 
 // src/local-apps.ts
-import { parseGGUFQuantLabel } from "@huggingface/gguf";
 function isAwqModel(model) {
   return model.config?.quantization_config?.quant_method === "awq";
 }
@@ -7852,6 +7916,9 @@ export {
   ALL_MODEL_LIBRARY_KEYS,
   DATASET_LIBRARIES_UI_ELEMENTS,
   DEFAULT_MEMORY_OPTIONS,
+  GGMLQuantizationType,
+  GGUF_QUANT_RE,
+  GGUF_QUANT_RE_GLOBAL,
   LIBRARY_TASK_MAPPING,
   LOCAL_APPS,
   MAPPING_DEFAULT_WIDGET,
@@ -7866,5 +7933,6 @@ export {
   SUBTASK_TYPES,
   TASKS_DATA,
   TASKS_MODEL_LIBRARIES,
+  parseGGUFQuantLabel,
   snippets_exports as snippets
 };