@huggingface/tasks 0.19.43 → 0.19.45
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/commonjs/local-apps.d.ts +1 -1
- package/dist/commonjs/local-apps.d.ts.map +1 -1
- package/dist/commonjs/local-apps.js +26 -7
- package/dist/commonjs/model-data.d.ts +4 -0
- package/dist/commonjs/model-data.d.ts.map +1 -1
- package/dist/commonjs/model-libraries-snippets.d.ts +1 -0
- package/dist/commonjs/model-libraries-snippets.d.ts.map +1 -1
- package/dist/commonjs/model-libraries-snippets.js +67 -2
- package/dist/commonjs/model-libraries.d.ts +8 -1
- package/dist/commonjs/model-libraries.d.ts.map +1 -1
- package/dist/commonjs/model-libraries.js +8 -1
- package/dist/commonjs/tasks/audio-text-to-text/data.d.ts +4 -0
- package/dist/commonjs/tasks/audio-text-to-text/data.d.ts.map +1 -0
- package/dist/commonjs/tasks/audio-text-to-text/data.js +67 -0
- package/dist/esm/local-apps.d.ts +1 -1
- package/dist/esm/local-apps.d.ts.map +1 -1
- package/dist/esm/local-apps.js +26 -7
- package/dist/esm/model-data.d.ts +4 -0
- package/dist/esm/model-data.d.ts.map +1 -1
- package/dist/esm/model-libraries-snippets.d.ts +1 -0
- package/dist/esm/model-libraries-snippets.d.ts.map +1 -1
- package/dist/esm/model-libraries-snippets.js +65 -1
- package/dist/esm/model-libraries.d.ts +8 -1
- package/dist/esm/model-libraries.d.ts.map +1 -1
- package/dist/esm/model-libraries.js +8 -1
- package/dist/esm/tasks/audio-text-to-text/data.d.ts +4 -0
- package/dist/esm/tasks/audio-text-to-text/data.d.ts.map +1 -0
- package/dist/esm/tasks/audio-text-to-text/data.js +65 -0
- package/package.json +1 -1
- package/src/local-apps.ts +28 -7
- package/src/model-data.ts +4 -0
- package/src/model-libraries-snippets.ts +87 -1
- package/src/model-libraries.ts +8 -1
- package/src/tasks/audio-text-to-text/about.md +139 -0
- package/src/tasks/audio-text-to-text/data.ts +70 -0
|
@@ -207,7 +207,7 @@ export declare const LOCAL_APPS: {
|
|
|
207
207
|
prettyLabel: string;
|
|
208
208
|
docsUrl: string;
|
|
209
209
|
mainTask: "text-generation";
|
|
210
|
-
displayOnModelPage:
|
|
210
|
+
displayOnModelPage: (model: ModelData) => boolean;
|
|
211
211
|
snippet: (model: ModelData, filepath?: string) => LocalAppSnippet[];
|
|
212
212
|
};
|
|
213
213
|
};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"local-apps.d.ts","sourceRoot":"","sources":["../../src/local-apps.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAKnD,MAAM,WAAW,eAAe;IAC/B;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;CAC3B;AAED;;GAEG;AACH,MAAM,MAAM,QAAQ,GAAG;IACtB;;OAEG;IACH,WAAW,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,QAAQ,EAAE,YAAY,CAAC;IACvB;;OAEG;IACH,SAAS,CAAC,EAAE,OAAO,CAAC;IAEpB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB;;OAEG;IACH,kBAAkB,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,OAAO,CAAC;CAClD,GAAG,CACD;IACA;;OAEG;IACH,QAAQ,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,MAAM,KAAK,GAAG,CAAC;CACtD,GACD;IACA;;;;OAIG;IACH,OAAO,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,MAAM,KAAK,MAAM,GAAG,MAAM,EAAE,GAAG,eAAe,GAAG,eAAe,EAAE,CAAC;CACzG,CACH,CAAC;AAqBF,iBAAS,UAAU,CAAC,KAAK,EAAE,SAAS,GAAG,OAAO,CAE7C;AAED,iBAAS,mBAAmB,CAAC,KAAK,EAAE,SAAS,WAE5C;
|
|
1
|
+
{"version":3,"file":"local-apps.d.ts","sourceRoot":"","sources":["../../src/local-apps.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAKnD,MAAM,WAAW,eAAe;IAC/B;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;CAC3B;AAED;;GAEG;AACH,MAAM,MAAM,QAAQ,GAAG;IACtB;;OAEG;IACH,WAAW,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,QAAQ,EAAE,YAAY,CAAC;IACvB;;OAEG;IACH,SAAS,CAAC,EAAE,OAAO,CAAC;IAEpB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB;;OAEG;IACH,kBAAkB,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,OAAO,CAAC;CAClD,GAAG,CACD;IACA;;OAEG;IACH,QAAQ,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,MAAM,KAAK,GAAG,CAAC;CACtD,GACD;IACA;;;;OAIG;IACH,OAAO,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,MAAM,KAAK,MAAM,GAAG,MAAM,EAAE,GAAG,eAAe,GAAG,eAAe,EAAE,CAAC;CACzG,CACH,CAAC;AAqBF,iBAAS,UAAU,CAAC,KAAK,EAAE,SAAS,GAAG,OAAO,CAE7C;AAED,iBAAS,mBAAmB,CAAC,KAAK,EAAE,SAAS,WAE5C;AAiRD;;;;;;;;;;GAUG;AACH,eAAO,MAAM,UAAU;;;;;;yBAvQS,SAAS,aAAa,MAAM,KAAG,eAAe,EAAE;;;;;;;yBAsCzC,SAAS,aAAa,MAAM,KAAG,eAAe,EAAE;;;;;;oCAoPzD,SAAS;yBA3MX,SAAS,KAAG,eAAe,EAAE;;;;;;;yBA4F5B,SAAS,KAAG,eAAe,EAAE;;;;;;;yBA7B/B,SAAS,KAAG,eAAe,EAAE;;;;;;;;;;;;;;yBAtFzB,SAAS,aAAa,MAAM,KAAG,eAAe,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;yBAJjD,SAAS,aAAa,MAAM,KAAG,MAAM;;;;;;;yBA0J1B,SAAS,aAAa,MAAM,KAAG,MAAM;;;;;;;yBAI9C,SAAS,aAAa,MAAM,KAAG,eAAe,EAAE;;CAkO5C,CAAC;AAErC,MAAM,MAAM,WAAW,GAAG,MAAM,OAAO,UAAU,CAAC"}
|
|
@@ -25,6 +25,9 @@ function isTgiModel(model) {
|
|
|
25
25
|
function isLlamaCppGgufModel(model) {
|
|
26
26
|
return !!model.gguf?.context_length;
|
|
27
27
|
}
|
|
28
|
+
function isAmdRyzenModel(model) {
|
|
29
|
+
return model.tags.includes("ryzenai-hybrid") || model.tags.includes("ryzenai-npu");
|
|
30
|
+
}
|
|
28
31
|
function isMlxModel(model) {
|
|
29
32
|
return model.tags.includes("mlx");
|
|
30
33
|
}
|
|
@@ -235,21 +238,37 @@ const snippetDockerModelRunner = (model, filepath) => {
|
|
|
235
238
|
};
|
|
236
239
|
const snippetLemonade = (model, filepath) => {
|
|
237
240
|
const tagName = getQuantTag(filepath);
|
|
238
|
-
const modelName = model.id.split("/")[1];
|
|
241
|
+
const modelName = model.id.includes("/") ? model.id.split("/")[1] : model.id;
|
|
242
|
+
// Get recipe according to model type
|
|
243
|
+
let simplifiedModelName;
|
|
244
|
+
let recipe;
|
|
245
|
+
let checkpoint;
|
|
246
|
+
let requirements;
|
|
247
|
+
if (model.tags.some((tag) => ["ryzenai-npu", "ryzenai-hybrid"].includes(tag))) {
|
|
248
|
+
recipe = model.tags.includes("ryzenai-npu") ? "oga-npu" : "oga-hybrid";
|
|
249
|
+
checkpoint = model.id;
|
|
250
|
+
requirements = " (requires RyzenAI 300 series)";
|
|
251
|
+
simplifiedModelName = modelName.split("-awq-")[0];
|
|
252
|
+
simplifiedModelName += recipe === "oga-npu" ? "-NPU" : "-Hybrid";
|
|
253
|
+
}
|
|
254
|
+
else {
|
|
255
|
+
recipe = "llamacpp";
|
|
256
|
+
checkpoint = `${model.id}${tagName}`;
|
|
257
|
+
requirements = "";
|
|
258
|
+
simplifiedModelName = modelName;
|
|
259
|
+
}
|
|
239
260
|
return [
|
|
240
261
|
{
|
|
241
262
|
title: "Pull the model",
|
|
242
263
|
setup: "# Download Lemonade from https://lemonade-server.ai/",
|
|
243
264
|
content: [
|
|
244
|
-
`lemonade-server pull user.${
|
|
245
|
-
--checkpoint ${model.id}${tagName} \\
|
|
246
|
-
--recipe llamacpp`,
|
|
265
|
+
`lemonade-server pull user.${simplifiedModelName} --checkpoint ${checkpoint} --recipe ${recipe}`,
|
|
247
266
|
"# Note: If you installed from source, use the lemonade-server-dev command instead.",
|
|
248
267
|
].join("\n"),
|
|
249
268
|
},
|
|
250
269
|
{
|
|
251
|
-
title:
|
|
252
|
-
content: `lemonade-server run user.${
|
|
270
|
+
title: `Run and chat with the model${requirements}`,
|
|
271
|
+
content: `lemonade-server run user.${simplifiedModelName}`,
|
|
253
272
|
},
|
|
254
273
|
{
|
|
255
274
|
title: "List all available models",
|
|
@@ -436,7 +455,7 @@ exports.LOCAL_APPS = {
|
|
|
436
455
|
prettyLabel: "Lemonade",
|
|
437
456
|
docsUrl: "https://lemonade-server.ai",
|
|
438
457
|
mainTask: "text-generation",
|
|
439
|
-
displayOnModelPage: isLlamaCppGgufModel,
|
|
458
|
+
displayOnModelPage: (model) => isLlamaCppGgufModel(model) || isAmdRyzenModel(model),
|
|
440
459
|
snippet: snippetLemonade,
|
|
441
460
|
},
|
|
442
461
|
};
|
|
@@ -43,6 +43,10 @@ export interface ModelData {
|
|
|
43
43
|
quant_method?: string;
|
|
44
44
|
};
|
|
45
45
|
tokenizer_config?: TokenizerConfig;
|
|
46
|
+
processor_config?: {
|
|
47
|
+
chat_template?: string;
|
|
48
|
+
};
|
|
49
|
+
chat_template_jinja?: string;
|
|
46
50
|
adapter_transformers?: {
|
|
47
51
|
model_name?: string;
|
|
48
52
|
model_class?: string;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"model-data.d.ts","sourceRoot":"","sources":["../../src/model-data.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACzD,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAE3D;;GAEG;AACH,MAAM,WAAW,SAAS;IACzB;;OAEG;IACH,EAAE,EAAE,MAAM,CAAC;IACX;;;OAGG;IACH,SAAS,EAAE,MAAM,CAAC;IAClB;;OAEG;IACH,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB;;OAEG;IACH,MAAM,CAAC,EAAE;QACR,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;QACzB;;WAEG;QACH,QAAQ,CAAC,EAAE;YACV;;eAEG;YACH,CAAC,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;SACpB,CAAC;QACF,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,mBAAmB,CAAC,EAAE;YACrB,IAAI,CAAC,EAAE,MAAM,CAAC;YACd,YAAY,CAAC,EAAE,OAAO,CAAC;YACvB,YAAY,CAAC,EAAE,OAAO,CAAC;YACvB;;eAEG;YACH,YAAY,CAAC,EAAE,MAAM,CAAC;SACtB,CAAC;QACF,gBAAgB,CAAC,EAAE,eAAe,CAAC;QACnC,oBAAoB,CAAC,EAAE;YACtB,UAAU,CAAC,EAAE,MAAM,CAAC;YACpB,WAAW,CAAC,EAAE,MAAM,CAAC;SACrB,CAAC;QACF,SAAS,CAAC,EAAE;YACX,WAAW,CAAC,EAAE,MAAM,CAAC;SACrB,CAAC;QACF,OAAO,CAAC,EAAE;YACT,KAAK,CAAC,EAAE;gBACP,IAAI,CAAC,EAAE,MAAM,CAAC;aACd,CAAC;YACF,YAAY,CAAC,EAAE,MAAM,CAAC;SACtB,CAAC;QACF,WAAW,CAAC,EAAE;YACb,qBAAqB,CAAC,EAAE,MAAM,CAAC;YAC/B,iBAAiB,CAAC,EAAE,MAAM,CAAC;YAC3B,gBAAgB,CAAC,EAAE,MAAM,CAAC;SAC1B,CAAC;QACF,IAAI,CAAC,EAAE;YACN,uBAAuB,CAAC,EAAE,MAAM,CAAC;YACjC,SAAS,CAAC,EAAE,MAAM,CAAC;SACnB,CAAC;QACF,SAAS,CAAC,EAAE;YACX,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;SACjB,CAAC;KACF,CAAC;IACF;;OAEG;IACH,IAAI,EAAE,MAAM,EAAE,CAAC;IACf;;OAEG;IACH,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;IACpC;;OAEG;IACH,YAAY,CAAC,EAAE,YAAY,GAAG,SAAS,CAAC;IACxC;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAChC;;;;;OAKG;IACH,UAAU,CAAC,EAAE,aAAa,EAAE,GAAG,SAAS,CAAC;IACzC;;;;;;;;;OASG;IACH,QAAQ,CAAC,EAAE;QACV,SAAS,CAAC,EACP,OAAO,GACP;YACA,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;SACpC,CAAC;QACL,UAAU,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;QAC/B,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;KAChC,CAAC;IACF;;;OAGG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,WAAW,CAAC,EAAE;QACb,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACnC,KAAK,EAAE,MAAM,CAAC;QACd,OAAO,EAAE,OAAO,CAAC;KACjB,CAAC;IACF,IAAI,CAAC,EAAE;QACN,KAAK,EAAE,MAAM,CAAC;QACd,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,cAAc,CAAC,EAAE,MAAM,CAAC;KACxB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;OAEG;IACH,UAAU,EAAE,MAAM,CAAC;IACnB;;OAEG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB;;OAEG;IACH,YAAY,CAAC,EAAE,YAAY,CAAC;IAC5B;;OAEG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;CACnB"}
|
|
1
|
+
{"version":3,"file":"model-data.d.ts","sourceRoot":"","sources":["../../src/model-data.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACzD,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAE3D;;GAEG;AACH,MAAM,WAAW,SAAS;IACzB;;OAEG;IACH,EAAE,EAAE,MAAM,CAAC;IACX;;;OAGG;IACH,SAAS,EAAE,MAAM,CAAC;IAClB;;OAEG;IACH,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB;;OAEG;IACH,MAAM,CAAC,EAAE;QACR,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;QACzB;;WAEG;QACH,QAAQ,CAAC,EAAE;YACV;;eAEG;YACH,CAAC,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;SACpB,CAAC;QACF,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,mBAAmB,CAAC,EAAE;YACrB,IAAI,CAAC,EAAE,MAAM,CAAC;YACd,YAAY,CAAC,EAAE,OAAO,CAAC;YACvB,YAAY,CAAC,EAAE,OAAO,CAAC;YACvB;;eAEG;YACH,YAAY,CAAC,EAAE,MAAM,CAAC;SACtB,CAAC;QACF,gBAAgB,CAAC,EAAE,eAAe,CAAC;QACnC,gBAAgB,CAAC,EAAE;YAClB,aAAa,CAAC,EAAE,MAAM,CAAC;SACvB,CAAC;QACF,mBAAmB,CAAC,EAAE,MAAM,CAAC;QAC7B,oBAAoB,CAAC,EAAE;YACtB,UAAU,CAAC,EAAE,MAAM,CAAC;YACpB,WAAW,CAAC,EAAE,MAAM,CAAC;SACrB,CAAC;QACF,SAAS,CAAC,EAAE;YACX,WAAW,CAAC,EAAE,MAAM,CAAC;SACrB,CAAC;QACF,OAAO,CAAC,EAAE;YACT,KAAK,CAAC,EAAE;gBACP,IAAI,CAAC,EAAE,MAAM,CAAC;aACd,CAAC;YACF,YAAY,CAAC,EAAE,MAAM,CAAC;SACtB,CAAC;QACF,WAAW,CAAC,EAAE;YACb,qBAAqB,CAAC,EAAE,MAAM,CAAC;YAC/B,iBAAiB,CAAC,EAAE,MAAM,CAAC;YAC3B,gBAAgB,CAAC,EAAE,MAAM,CAAC;SAC1B,CAAC;QACF,IAAI,CAAC,EAAE;YACN,uBAAuB,CAAC,EAAE,MAAM,CAAC;YACjC,SAAS,CAAC,EAAE,MAAM,CAAC;SACnB,CAAC;QACF,SAAS,CAAC,EAAE;YACX,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;SACjB,CAAC;KACF,CAAC;IACF;;OAEG;IACH,IAAI,EAAE,MAAM,EAAE,CAAC;IACf;;OAEG;IACH,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;IACpC;;OAEG;IACH,YAAY,CAAC,EAAE,YAAY,GAAG,SAAS,CAAC;IACxC;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAChC;;;;;OAKG;IACH,UAAU,CAAC,EAAE,aAAa,EAAE,GAAG,SAAS,CAAC;IACzC;;;;;;;;;OASG;IACH,QAAQ,CAAC,EAAE;QACV,SAAS,CAAC,EACP,OAAO,GACP;YACA,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;SACpC,CAAC;QACL,UAAU,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;QAC/B,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;KAChC,CAAC;IACF;;;OAGG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,WAAW,CAAC,EAAE;QACb,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACnC,KAAK,EAAE,MAAM,CAAC;QACd,OAAO,EAAE,OAAO,CAAC;KACjB,CAAC;IACF,IAAI,CAAC,EAAE;QACN,KAAK,EAAE,MAAM,CAAC;QACd,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,cAAc,CAAC,EAAE,MAAM,CAAC;KACxB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;OAEG;IACH,UAAU,EAAE,MAAM,CAAC;IACnB;;OAEG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB;;OAEG;IACH,YAAY,CAAC,EAAE,YAAY,CAAC;IAC5B;;OAEG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;CACnB"}
|
|
@@ -86,6 +86,7 @@ export declare const univa: (model: ModelData) => string[];
|
|
|
86
86
|
export declare const mlxim: (model: ModelData) => string[];
|
|
87
87
|
export declare const mlx: (model: ModelData) => string[];
|
|
88
88
|
export declare const model2vec: (model: ModelData) => string[];
|
|
89
|
+
export declare const pruna: (model: ModelData) => string[];
|
|
89
90
|
export declare const nemo: (model: ModelData) => string[];
|
|
90
91
|
export declare const outetts: (model: ModelData) => string[];
|
|
91
92
|
export declare const pxia: (model: ModelData) => string[];
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"model-libraries-snippets.d.ts","sourceRoot":"","sources":["../../src/model-libraries-snippets.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAkBjD,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAKjD,CAAC;AAkBF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAKjD,CAAC;AAEF,eAAO,MAAM,OAAO,GAAI,OAAO,SAAS,KAAG,MAAM,EAIhD,CAAC;AAEF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAkBlD,CAAC;AAaF,eAAO,MAAM,IAAI,GAAI,OAAO,SAAS,KAAG,MAAM,EAY7C,CAAC;AAEF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAenC,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAsDnC,CAAC;AAEF,eAAO,MAAM,cAAc,QAAO,MAAM,EAcvC,CAAC;AAEF,eAAO,MAAM,iBAAiB,GAAI,OAAO,SAAS,KAAG,MAAM,EA6C1D,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAwBlD,CAAC;AAEF,eAAO,MAAM,eAAe,QAAO,MAAM,EAoBxC,CAAC;AAEF,eAAO,MAAM,GAAG,GAAI,OAAO,SAAS,KAAG,MAAM,EAS5C,CAAC;AAEF,eAAO,MAAM,iBAAiB,GAAI,OAAO,SAAS,KAAG,MAAM,EAY1D,CAAC;AAkKF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EA6BlD,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,OAAO,SAAS,KAAG,MAAM,EAwCrD,CAAC;AAEF,eAAO,MAAM,gBAAgB,GAAI,OAAO,SAAS,KAAG,MAAM,EAgBzD,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,OAAO,SAAS,KAAG,MAAM,EAmBrD,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAgB/C,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAMlD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EASlD,CAAC;AAIF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAO/C,CAAC;AAEF,eAAO,MAAM,OAAO,GAAI,OAAO,SAAS,KAAG,MAAM,EAMhD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAgBjD,CAAC;AAEF,eAAO,MAAM,OAAO,GAAI,OAAO,SAAS,KAAG,MAAM,EAehD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAS9C,CAAC;AA4EF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAsBlD,CAAC;AAEF,eAAO,MAAM,UAAU,GAAI,OAAO,SAAS,KAAG,MAAM,EA+BnD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EASlD,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,OAAO,SAAS,KAAG,MAAM,EA+BrD,CAAC;AAEF,eAAO,MAAM,gBAAgB,GAAI,OAAO,SAAS,KAAG,MAAM,EA4BzD,CAAC;AAEF,eAAO,MAAM,OAAO,GAAI,OAAO,SAAS,KAAG,MAAM,EAuChD,CAAC;AAEF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAOjD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAIlD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAKlD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAQlD,CAAC;AAEF,eAAO,MAAM,aAAa,QAAO,MAAM,EAQtC,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAKlD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAsBlD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAmDlD,CAAC;AAEF,eAAO,MAAM,kBAAkB,GAAI,OAAO,SAAS,KAAG,MAAM,EAgB3D,CAAC;AACF,eAAO,MAAM,WAAW,GAAI,OAAO,SAAS,KAAG,MAAM,EAUpD,CAAC;AAEF,eAAO,MAAM,uBAAuB,GAAI,OAAO,SAAS,KAAG,MAAM,EAehE,CAAC;AAiBF,eAAO,MAAM,cAAc,GAAI,OAAO,SAAS,KAAG,MAAM,EAKvD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,OAAO,SAAS,KAAG,MAAM,EAKrD,CAAC;AAyBF,eAAO,MAAM,aAAa,GAAI,OAAO,SAAS,KAAG,MAAM,EAOtD,CAAC;AAEF,eAAO,MAAM,IAAI,GAAI,OAAO,SAAS,KAAG,MAAM,EAI7C,CAAC;AAEF,eAAO,MAAM,OAAO,QAA6B,MAAM,EAQtD,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAanC,CAAC;AAsCF,eAAO,MAAM,OAAO,GAAI,OAAO,SAAS,KAAG,MAAM,EAehD,CAAC;AAEF,eAAO,MAAM,kBAAkB,GAAI,OAAO,SAAS,KAAG,MAAM,EAmC3D,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,IAAI,GAAI,OAAO,SAAS,KAAG,MAAM,EA2B7C,CAAC;AAEF,eAAO,MAAM,aAAa,GAAI,OAAO,SAAS,KAAG,MAAM,EAEtD,CAAC;AASF,eAAO,MAAM,oBAAoB,GAAI,OAAO,SAAS,KAAG,MAAM,EAuC7D,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAU9C,CAAC;AAEF,eAAO,MAAM,WAAW,GAAI,OAAO,SAAS,KAAG,MAAM,EAIpD,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAK/C,CAAC;AAkBF,eAAO,MAAM,WAAW,GAAI,OAAO,SAAS,KAAG,MAAM,EAkBpD,CAAC;AAEF,eAAO,MAAM,UAAU,GAAI,OAAO,SAAS,KAAG,MAAM,EAInD,CAAC;
|
|
1
|
+
{"version":3,"file":"model-libraries-snippets.d.ts","sourceRoot":"","sources":["../../src/model-libraries-snippets.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAkBjD,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAKjD,CAAC;AAkBF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAKjD,CAAC;AAEF,eAAO,MAAM,OAAO,GAAI,OAAO,SAAS,KAAG,MAAM,EAIhD,CAAC;AAEF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAkBlD,CAAC;AAaF,eAAO,MAAM,IAAI,GAAI,OAAO,SAAS,KAAG,MAAM,EAY7C,CAAC;AAEF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAenC,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAsDnC,CAAC;AAEF,eAAO,MAAM,cAAc,QAAO,MAAM,EAcvC,CAAC;AAEF,eAAO,MAAM,iBAAiB,GAAI,OAAO,SAAS,KAAG,MAAM,EA6C1D,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAwBlD,CAAC;AAEF,eAAO,MAAM,eAAe,QAAO,MAAM,EAoBxC,CAAC;AAEF,eAAO,MAAM,GAAG,GAAI,OAAO,SAAS,KAAG,MAAM,EAS5C,CAAC;AAEF,eAAO,MAAM,iBAAiB,GAAI,OAAO,SAAS,KAAG,MAAM,EAY1D,CAAC;AAkKF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EA6BlD,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,OAAO,SAAS,KAAG,MAAM,EAwCrD,CAAC;AAEF,eAAO,MAAM,gBAAgB,GAAI,OAAO,SAAS,KAAG,MAAM,EAgBzD,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,OAAO,SAAS,KAAG,MAAM,EAmBrD,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAgB/C,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAMlD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EASlD,CAAC;AAIF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAO/C,CAAC;AAEF,eAAO,MAAM,OAAO,GAAI,OAAO,SAAS,KAAG,MAAM,EAMhD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAgBjD,CAAC;AAEF,eAAO,MAAM,OAAO,GAAI,OAAO,SAAS,KAAG,MAAM,EAehD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAS9C,CAAC;AA4EF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAsBlD,CAAC;AAEF,eAAO,MAAM,UAAU,GAAI,OAAO,SAAS,KAAG,MAAM,EA+BnD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EASlD,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,OAAO,SAAS,KAAG,MAAM,EA+BrD,CAAC;AAEF,eAAO,MAAM,gBAAgB,GAAI,OAAO,SAAS,KAAG,MAAM,EA4BzD,CAAC;AAEF,eAAO,MAAM,OAAO,GAAI,OAAO,SAAS,KAAG,MAAM,EAuChD,CAAC;AAEF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAOjD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAIlD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAKlD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAQlD,CAAC;AAEF,eAAO,MAAM,aAAa,QAAO,MAAM,EAQtC,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAKlD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAsBlD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAmDlD,CAAC;AAEF,eAAO,MAAM,kBAAkB,GAAI,OAAO,SAAS,KAAG,MAAM,EAgB3D,CAAC;AACF,eAAO,MAAM,WAAW,GAAI,OAAO,SAAS,KAAG,MAAM,EAUpD,CAAC;AAEF,eAAO,MAAM,uBAAuB,GAAI,OAAO,SAAS,KAAG,MAAM,EAehE,CAAC;AAiBF,eAAO,MAAM,cAAc,GAAI,OAAO,SAAS,KAAG,MAAM,EAKvD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,OAAO,SAAS,KAAG,MAAM,EAKrD,CAAC;AAyBF,eAAO,MAAM,aAAa,GAAI,OAAO,SAAS,KAAG,MAAM,EAOtD,CAAC;AAEF,eAAO,MAAM,IAAI,GAAI,OAAO,SAAS,KAAG,MAAM,EAI7C,CAAC;AAEF,eAAO,MAAM,OAAO,QAA6B,MAAM,EAQtD,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAanC,CAAC;AAsCF,eAAO,MAAM,OAAO,GAAI,OAAO,SAAS,KAAG,MAAM,EAehD,CAAC;AAEF,eAAO,MAAM,kBAAkB,GAAI,OAAO,SAAS,KAAG,MAAM,EAmC3D,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,IAAI,GAAI,OAAO,SAAS,KAAG,MAAM,EA2B7C,CAAC;AAEF,eAAO,MAAM,aAAa,GAAI,OAAO,SAAS,KAAG,MAAM,EAEtD,CAAC;AASF,eAAO,MAAM,oBAAoB,GAAI,OAAO,SAAS,KAAG,MAAM,EAuC7D,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAU9C,CAAC;AAEF,eAAO,MAAM,WAAW,GAAI,OAAO,SAAS,KAAG,MAAM,EAIpD,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAK/C,CAAC;AAkBF,eAAO,MAAM,WAAW,GAAI,OAAO,SAAS,KAAG,MAAM,EAkBpD,CAAC;AAEF,eAAO,MAAM,UAAU,GAAI,OAAO,SAAS,KAAG,MAAM,EAInD,CAAC;AAOF,eAAO,MAAM,YAAY,GAAI,OAAO,SAAS,KAAG,MAAM,EAyGrD,CAAC;AAEF,eAAO,MAAM,cAAc,GAAI,OAAO,SAAS,KAAG,MAAM,EAcvD,CAAC;AAiBF,eAAO,MAAM,IAAI,GAAI,OAAO,SAAS,KAAG,MAAM,EAiB7C,CAAC;AAEF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAKjD,CAAC;AAEF,eAAO,MAAM,gBAAgB,GAAI,OAAO,SAAS,KAAG,MAAM,EAMzD,CAAC;AAgBF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAEjD,CAAC;AAEF,eAAO,MAAM,MAAM,QAA6B,MAAM,EAMrD,CAAC;AAEF,eAAO,MAAM,IAAI,GAAI,OAAO,SAAS,KAAG,MAAM,EAkB7C,CAAC;AAEF,eAAO,MAAM,UAAU,GAAI,OAAO,SAAS,KAAG,MAAM,EAWnD,CAAC;AAEF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAQ/C,CAAC;AAEF,eAAO,MAAM,UAAU,GAAI,OAAO,SAAS,KAAG,MAAM,EAInD,CAAC;AAEF,eAAO,MAAM,GAAG,QAAO,MAAM,EAgB5B,CAAC;AAEF,eAAO,MAAM,OAAO,QAAO,MAAM,EAYhC,CAAC;AAEF,eAAO,MAAM,WAAW,GAAI,OAAO,SAAS,KAAG,MAAM,EAiBpD,CAAC;AAEF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAYjD,CAAC;AAEF,eAAO,MAAM,WAAW,GAAI,OAAO,SAAS,KAAG,MAAM,EAKpD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAW9C,CAAC;AAmEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,GAAG,GAAI,OAAO,SAAS,KAAG,MAAM,EAY5C,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAIlD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EA0B9C,CAAC;AAuDF,eAAO,MAAM,IAAI,GAAI,OAAO,SAAS,KAAG,MAAM,EAQ7C,CAAC;AAEF,eAAO,MAAM,OAAO,GAAI,OAAO,SAAS,KAAG,MAAM,EAuBhD,CAAC;AAEF,eAAO,MAAM,IAAI,GAAI,OAAO,SAAS,KAAG,MAAM,EAI7C,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAI/C,CAAC;AA4BF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAO/C,CAAC;AAEF,eAAO,MAAM,UAAU,GAAI,OAAO,SAAS,KAAG,MAAM,EAUnD,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAYnC,CAAC;AAEF,eAAO,MAAM,cAAc,GAAI,OAAO,SAAS,KAAG,MAAM,EAKvD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAiB9C,CAAC"}
|
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
exports.renderformer = exports.relik = exports.pyannote_audio = exports.pyannote_audio_pipeline = exports.phantom_wan = exports.perception_encoder = exports.paddleocr = exports.paddlenlp = exports.open_clip = exports.mesh_anything = exports.matanyone = exports.mars5_tts = exports.mamba_ssm = exports.tf_keras = exports.lerobot = exports.llama_cpp_python = exports.lightning_ir = exports.kittentts = exports.kimi_audio = exports.keras_hub = exports.keras = exports.htrflow = exports.indextts = exports.gliner = exports.flair = exports.fairseq = exports.espnet = exports.espnetASR = exports.espnetTTS = exports.edsnlp = exports.cartesia_mlx = exports.cartesia_pytorch = exports.diffusionkit = exports.diffusers = exports.describe_anything = exports.dia = exports.derm_foundation = exports.depth_pro = exports.depth_anything_v2 = exports.cxr_foundation = exports.contexttab = exports.chatterbox = exports.bm25s = exports.bertopic = exports.ben2 = exports.audioseal = exports.asteroid = exports.araclip = exports.allennlp = exports.adapters = void 0;
|
|
4
|
-
exports.zonos = exports.hezar = exports.threedtopia_xl = exports.whisperkit = exports.audiocraft = exports.anemoi = exports.pythae = exports.pxia = exports.outetts = exports.nemo = exports.model2vec = exports.mlx = exports.mlxim = exports.univa = exports.swarmformer = exports.birefnet = exports.ultralytics = exports.chattts = exports.vui = exports.voicecraft = exports.lvface = exports.vfimamba = exports.videoprism = exports.sana = exports.sentis = exports.mlAgents = exports.stableBaselines3 = exports.fasttext = exports.peft = exports.transformersJS = exports.transformers = exports.terratorch = exports.speechbrain = exports.stanza = exports.span_marker = exports.spacy = exports.setfit = exports.sentenceTransformers = exports.sampleFactory = exports.sam2 = exports.fastai = exports.stable_audio_tools = exports.sklearn = exports.seed_story = exports.saelens = exports.timm = exports.tensorflowtts = void 0;
|
|
4
|
+
exports.zonos = exports.hezar = exports.threedtopia_xl = exports.whisperkit = exports.audiocraft = exports.anemoi = exports.pythae = exports.pxia = exports.outetts = exports.nemo = exports.pruna = exports.model2vec = exports.mlx = exports.mlxim = exports.univa = exports.swarmformer = exports.birefnet = exports.ultralytics = exports.chattts = exports.vui = exports.voicecraft = exports.lvface = exports.vfimamba = exports.videoprism = exports.sana = exports.sentis = exports.mlAgents = exports.stableBaselines3 = exports.fasttext = exports.peft = exports.transformersJS = exports.transformers = exports.terratorch = exports.speechbrain = exports.stanza = exports.span_marker = exports.spacy = exports.setfit = exports.sentenceTransformers = exports.sampleFactory = exports.sam2 = exports.fastai = exports.stable_audio_tools = exports.sklearn = exports.seed_story = exports.saelens = exports.timm = exports.tensorflowtts = void 0;
|
|
5
5
|
const library_to_tasks_js_1 = require("./library-to-tasks.js");
|
|
6
6
|
const inputs_js_1 = require("./snippets/inputs.js");
|
|
7
7
|
const common_js_1 = require("./snippets/common.js");
|
|
@@ -1427,6 +1427,9 @@ const terratorch = (model) => [
|
|
|
1427
1427
|
model = BACKBONE_REGISTRY.build("${model.id}")`,
|
|
1428
1428
|
];
|
|
1429
1429
|
exports.terratorch = terratorch;
|
|
1430
|
+
const hasChatTemplate = (model) => model.config?.tokenizer_config?.chat_template !== undefined ||
|
|
1431
|
+
model.config?.processor_config?.chat_template !== undefined ||
|
|
1432
|
+
model.config?.chat_template_jinja !== undefined;
|
|
1430
1433
|
const transformers = (model) => {
|
|
1431
1434
|
const info = model.transformersInfo;
|
|
1432
1435
|
if (!info) {
|
|
@@ -1441,7 +1444,7 @@ const transformers = (model) => {
|
|
|
1441
1444
|
? "extractor"
|
|
1442
1445
|
: "processor";
|
|
1443
1446
|
autoSnippet.push("# Load model directly", `from transformers import ${info.processor}, ${info.auto_model}`, "", `${processorVarName} = ${info.processor}.from_pretrained("${model.id}"` + remote_code_snippet + ")", `model = ${info.auto_model}.from_pretrained("${model.id}"` + remote_code_snippet + ")");
|
|
1444
|
-
if (model.tags.includes("conversational")) {
|
|
1447
|
+
if (model.tags.includes("conversational") && hasChatTemplate(model)) {
|
|
1445
1448
|
if (model.tags.includes("image-text-to-text")) {
|
|
1446
1449
|
autoSnippet.push("messages = [", [
|
|
1447
1450
|
" {",
|
|
@@ -1810,6 +1813,68 @@ const model2vec = (model) => [
|
|
|
1810
1813
|
model = StaticModel.from_pretrained("${model.id}")`,
|
|
1811
1814
|
];
|
|
1812
1815
|
exports.model2vec = model2vec;
|
|
1816
|
+
const pruna = (model) => {
|
|
1817
|
+
let snippets;
|
|
1818
|
+
if (model.tags.includes("diffusers")) {
|
|
1819
|
+
snippets = pruna_diffusers(model);
|
|
1820
|
+
}
|
|
1821
|
+
else if (model.tags.includes("transformers")) {
|
|
1822
|
+
snippets = pruna_transformers(model);
|
|
1823
|
+
}
|
|
1824
|
+
else {
|
|
1825
|
+
snippets = pruna_default(model);
|
|
1826
|
+
}
|
|
1827
|
+
const ensurePrunaModelImport = (snippet) => {
|
|
1828
|
+
if (!/^from pruna import PrunaModel/m.test(snippet)) {
|
|
1829
|
+
return `from pruna import PrunaModel\n${snippet}`;
|
|
1830
|
+
}
|
|
1831
|
+
return snippet;
|
|
1832
|
+
};
|
|
1833
|
+
snippets = snippets.map(ensurePrunaModelImport);
|
|
1834
|
+
if (model.tags.includes("pruna_pro-ai")) {
|
|
1835
|
+
return snippets.map((snippet) => snippet.replace(/\bpruna\b/g, "pruna_pro").replace(/\bPrunaModel\b/g, "PrunaProModel"));
|
|
1836
|
+
}
|
|
1837
|
+
return snippets;
|
|
1838
|
+
};
|
|
1839
|
+
exports.pruna = pruna;
|
|
1840
|
+
const pruna_diffusers = (model) => {
|
|
1841
|
+
const diffusersSnippets = (0, exports.diffusers)(model);
|
|
1842
|
+
return diffusersSnippets.map((snippet) => snippet
|
|
1843
|
+
// Replace pipeline classes with PrunaModel
|
|
1844
|
+
.replace(/\b\w*Pipeline\w*\b/g, "PrunaModel")
|
|
1845
|
+
// Clean up diffusers imports containing PrunaModel
|
|
1846
|
+
.replace(/from diffusers import ([^,\n]*PrunaModel[^,\n]*)/g, "")
|
|
1847
|
+
.replace(/from diffusers import ([^,\n]+),?\s*([^,\n]*PrunaModel[^,\n]*)/g, "from diffusers import $1")
|
|
1848
|
+
.replace(/from diffusers import\s*(\n|$)/g, "")
|
|
1849
|
+
// Fix PrunaModel imports
|
|
1850
|
+
.replace(/from diffusers import PrunaModel/g, "from pruna import PrunaModel")
|
|
1851
|
+
.replace(/from diffusers import ([^,\n]+), PrunaModel/g, "from diffusers import $1")
|
|
1852
|
+
.replace(/from diffusers import PrunaModel, ([^,\n]+)/g, "from diffusers import $1")
|
|
1853
|
+
// Clean up whitespace
|
|
1854
|
+
.replace(/\n\n+/g, "\n")
|
|
1855
|
+
.trim());
|
|
1856
|
+
};
|
|
1857
|
+
const pruna_transformers = (model) => {
|
|
1858
|
+
const info = model.transformersInfo;
|
|
1859
|
+
const transformersSnippets = (0, exports.transformers)(model);
|
|
1860
|
+
// Replace pipeline with PrunaModel
|
|
1861
|
+
let processedSnippets = transformersSnippets.map((snippet) => snippet
|
|
1862
|
+
.replace(/from transformers import pipeline/g, "from pruna import PrunaModel")
|
|
1863
|
+
.replace(/pipeline\([^)]*\)/g, `PrunaModel.from_pretrained("${model.id}")`));
|
|
1864
|
+
// Additional cleanup if auto_model info is available
|
|
1865
|
+
if (info?.auto_model) {
|
|
1866
|
+
processedSnippets = processedSnippets.map((snippet) => snippet
|
|
1867
|
+
.replace(new RegExp(`from transformers import ${info.auto_model}\n?`, "g"), "")
|
|
1868
|
+
.replace(new RegExp(`${info.auto_model}.from_pretrained`, "g"), "PrunaModel.from_pretrained")
|
|
1869
|
+
.replace(new RegExp(`^.*from.*import.*(, *${info.auto_model})+.*$`, "gm"), (line) => line.replace(new RegExp(`, *${info.auto_model}`, "g"), "")));
|
|
1870
|
+
}
|
|
1871
|
+
return processedSnippets;
|
|
1872
|
+
};
|
|
1873
|
+
const pruna_default = (model) => [
|
|
1874
|
+
`from pruna import PrunaModel
|
|
1875
|
+
model = PrunaModel.from_pretrained("${model.id}")
|
|
1876
|
+
`,
|
|
1877
|
+
];
|
|
1813
1878
|
const nemo = (model) => {
|
|
1814
1879
|
let command = undefined;
|
|
1815
1880
|
// Resolve the tag to a nemo domain/sub-domain
|
|
@@ -792,6 +792,13 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
|
|
|
792
792
|
filter: false;
|
|
793
793
|
countDownloads: string;
|
|
794
794
|
};
|
|
795
|
+
"pruna-ai": {
|
|
796
|
+
prettyLabel: string;
|
|
797
|
+
repoName: string;
|
|
798
|
+
repoUrl: string;
|
|
799
|
+
snippets: (model: ModelData) => string[];
|
|
800
|
+
docsUrl: string;
|
|
801
|
+
};
|
|
795
802
|
pxia: {
|
|
796
803
|
prettyLabel: string;
|
|
797
804
|
repoName: string;
|
|
@@ -1247,5 +1254,5 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
|
|
|
1247
1254
|
};
|
|
1248
1255
|
export type ModelLibraryKey = keyof typeof MODEL_LIBRARIES_UI_ELEMENTS;
|
|
1249
1256
|
export declare const ALL_MODEL_LIBRARY_KEYS: ModelLibraryKey[];
|
|
1250
|
-
export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("acestep" | "adapter-transformers" | "allennlp" | "anemoi" | "araclip" | "asteroid" | "audiocraft" | "audioseal" | "bagel-mot" | "bboxmaskpose" | "ben2" | "bertopic" | "big_vision" | "birder" | "birefnet" | "bm25s" | "champ" | "chatterbox" | "chat_tts" | "colpali" | "comet" | "contexttab" | "cosmos" | "cxr-foundation" | "deepforest" | "depth-anything-v2" | "depth-pro" | "derm-foundation" | "describe-anything" | "dia-tts" | "diffree" | "diffusers" | "diffusionkit" | "doctr" | "cartesia_pytorch" | "cartesia_mlx" | "clipscope" | "cosyvoice" | "cotracker" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "fme" | "gemma.cpp" | "geometry-crafter" | "gliner" | "glyph-byt5" | "grok" | "hallo" | "hermes" | "hezar" | "htrflow" | "hunyuan-dit" | "hunyuan3d-2" | "hunyuanworld-voyager" | "imstoucan" | "index-tts" | "infinitetalk" | "infinite-you" | "keras" | "tf-keras" | "keras-hub" | "kimi-audio" | "kittentts" | "kronos" | "k2" | "lightning-ir" | "litert-lm" | "lerobot" | "liveportrait" | "llama-cpp-python" | "mini-omni2" | "mindspore" | "magi-1" | "magenta-realtime" | "mamba-ssm" | "mars5-tts" | "matanyone" | "mesh-anything" | "merlin" | "medvae" | "mitie" | "ml-agents" | "mlx" | "mlx-image" | "mlc-llm" | "model2vec" | "moshi" | "mtvcraft" | "nemo" | "open-oasis" | "open_clip" | "open-sora" | "outetts" | "paddlenlp" | "PaddleOCR" | "peft" | "perception-encoder" | "phantom-wan" | "pxia" | "pyannote-audio" | "py-feat" | "pythae" | "recurrentgemma" | "relik" | "refiners" | "renderformer" | "reverb" | "rkllm" | "saelens" | "sam2" | "sample-factory" | "sapiens" | "seedvr" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "ssr-speech" | "stable-audio-tools" | "monkeyocr" | "diffusion-single-file" | "seed-story" | "soloaudio" | "songbloom" | "stable-baselines3" | "stanza" | "swarmformer" | "f5-tts" | "genmo" | "tencent-song-generation" | "tensorflowtts" | "tabpfn" | "terratorch" | "tic-clip" | "timesfm" | "timm" | "tirex" | "torchgeo" | "transformers" | "transformers.js" | "trellis" | "ultralytics" | "univa" | "uni-3dar" | "unity-sentis" | "sana" | "videoprism" | "vfi-mamba" | "lvface" | "voicecraft" | "vui" | "wan2.2" | "wham" | "whisperkit" | "yolov10" | "zonos" | "3dtopia-xl")[];
|
|
1257
|
+
export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("acestep" | "adapter-transformers" | "allennlp" | "anemoi" | "araclip" | "asteroid" | "audiocraft" | "audioseal" | "bagel-mot" | "bboxmaskpose" | "ben2" | "bertopic" | "big_vision" | "birder" | "birefnet" | "bm25s" | "champ" | "chatterbox" | "chat_tts" | "colpali" | "comet" | "contexttab" | "cosmos" | "cxr-foundation" | "deepforest" | "depth-anything-v2" | "depth-pro" | "derm-foundation" | "describe-anything" | "dia-tts" | "diffree" | "diffusers" | "diffusionkit" | "doctr" | "cartesia_pytorch" | "cartesia_mlx" | "clipscope" | "cosyvoice" | "cotracker" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "fme" | "gemma.cpp" | "geometry-crafter" | "gliner" | "glyph-byt5" | "grok" | "hallo" | "hermes" | "hezar" | "htrflow" | "hunyuan-dit" | "hunyuan3d-2" | "hunyuanworld-voyager" | "imstoucan" | "index-tts" | "infinitetalk" | "infinite-you" | "keras" | "tf-keras" | "keras-hub" | "kimi-audio" | "kittentts" | "kronos" | "k2" | "lightning-ir" | "litert-lm" | "lerobot" | "liveportrait" | "llama-cpp-python" | "mini-omni2" | "mindspore" | "magi-1" | "magenta-realtime" | "mamba-ssm" | "mars5-tts" | "matanyone" | "mesh-anything" | "merlin" | "medvae" | "mitie" | "ml-agents" | "mlx" | "mlx-image" | "mlc-llm" | "model2vec" | "moshi" | "mtvcraft" | "nemo" | "open-oasis" | "open_clip" | "open-sora" | "outetts" | "paddlenlp" | "PaddleOCR" | "peft" | "perception-encoder" | "phantom-wan" | "pruna-ai" | "pxia" | "pyannote-audio" | "py-feat" | "pythae" | "recurrentgemma" | "relik" | "refiners" | "renderformer" | "reverb" | "rkllm" | "saelens" | "sam2" | "sample-factory" | "sapiens" | "seedvr" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "ssr-speech" | "stable-audio-tools" | "monkeyocr" | "diffusion-single-file" | "seed-story" | "soloaudio" | "songbloom" | "stable-baselines3" | "stanza" | "swarmformer" | "f5-tts" | "genmo" | "tencent-song-generation" | "tensorflowtts" | "tabpfn" | "terratorch" | "tic-clip" | "timesfm" | "timm" | "tirex" | "torchgeo" | "transformers" | "transformers.js" | "trellis" | "ultralytics" | "univa" | "uni-3dar" | "unity-sentis" | "sana" | "videoprism" | "vfi-mamba" | "lvface" | "voicecraft" | "vui" | "wan2.2" | "wham" | "whisperkit" | "yolov10" | "zonos" | "3dtopia-xl")[];
|
|
1251
1258
|
//# sourceMappingURL=model-libraries.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"model-libraries.d.ts","sourceRoot":"","sources":["../../src/model-libraries.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AAEzE;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;;;OAIG;IACH,WAAW,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,MAAM,EAAE,CAAC;IAC1C;;;;;OAKG;IACH,cAAc,CAAC,EAAE,kBAAkB,CAAC;IACpC;;;OAGG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;CACjB;AAED;;;;;;;;;;;;;GAaG;AAEH,eAAO,MAAM,2BAA2B
|
|
1
|
+
{"version":3,"file":"model-libraries.d.ts","sourceRoot":"","sources":["../../src/model-libraries.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AAEzE;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;;;OAIG;IACH,WAAW,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,MAAM,EAAE,CAAC;IAC1C;;;;;OAKG;IACH,cAAc,CAAC,EAAE,kBAAkB,CAAC;IACpC;;;OAGG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;CACjB;AAED;;;;;;;;;;;;;GAaG;AAEH,eAAO,MAAM,2BAA2B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAorCI,CAAC;AAE7C,MAAM,MAAM,eAAe,GAAG,MAAM,OAAO,2BAA2B,CAAC;AAEvE,eAAO,MAAM,sBAAsB,EAA+C,eAAe,EAAE,CAAC;AAEpG,eAAO,MAAM,8BAA8B,ivEAQ1B,CAAC"}
|
|
@@ -793,6 +793,13 @@ exports.MODEL_LIBRARIES_UI_ELEMENTS = {
|
|
|
793
793
|
filter: false,
|
|
794
794
|
countDownloads: `path_extension:"pth"`,
|
|
795
795
|
},
|
|
796
|
+
"pruna-ai": {
|
|
797
|
+
prettyLabel: "Pruna AI",
|
|
798
|
+
repoName: "Pruna AI",
|
|
799
|
+
repoUrl: "https://github.com/PrunaAI/pruna",
|
|
800
|
+
snippets: snippets.pruna,
|
|
801
|
+
docsUrl: "https://docs.pruna.ai",
|
|
802
|
+
},
|
|
796
803
|
pxia: {
|
|
797
804
|
prettyLabel: "pxia",
|
|
798
805
|
repoName: "pxia",
|
|
@@ -1064,7 +1071,7 @@ exports.MODEL_LIBRARIES_UI_ELEMENTS = {
|
|
|
1064
1071
|
repoUrl: "https://github.com/IBM/terratorch",
|
|
1065
1072
|
docsUrl: "https://ibm.github.io/terratorch/",
|
|
1066
1073
|
filter: false,
|
|
1067
|
-
countDownloads: `path_extension:"pt"`,
|
|
1074
|
+
countDownloads: `path_extension:"pt" OR path_extension:"ckpt"`,
|
|
1068
1075
|
snippets: snippets.terratorch,
|
|
1069
1076
|
},
|
|
1070
1077
|
"tic-clip": {
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"data.d.ts","sourceRoot":"","sources":["../../../../src/tasks/audio-text-to-text/data.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAElD,QAAA,MAAM,QAAQ,EAAE,cAiEf,CAAC;AAEF,eAAe,QAAQ,CAAC"}
|
|
@@ -0,0 +1,67 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
const taskData = {
|
|
4
|
+
datasets: [
|
|
5
|
+
{
|
|
6
|
+
description: "A dataset containing audio conversations with question–answer pairs.",
|
|
7
|
+
id: "nvidia/AF-Think",
|
|
8
|
+
},
|
|
9
|
+
{
|
|
10
|
+
description: "A more advanced and comprehensive dataset that contains characteristics of the audio as well",
|
|
11
|
+
id: "tsinghua-ee/QualiSpeech",
|
|
12
|
+
},
|
|
13
|
+
],
|
|
14
|
+
demo: {
|
|
15
|
+
inputs: [
|
|
16
|
+
{
|
|
17
|
+
filename: "audio.wav",
|
|
18
|
+
type: "audio",
|
|
19
|
+
},
|
|
20
|
+
{
|
|
21
|
+
label: "Text Prompt",
|
|
22
|
+
content: "What is the gender of the speaker?",
|
|
23
|
+
type: "text",
|
|
24
|
+
},
|
|
25
|
+
],
|
|
26
|
+
outputs: [
|
|
27
|
+
{
|
|
28
|
+
label: "Generated Text",
|
|
29
|
+
content: "The gender of the speaker is female.",
|
|
30
|
+
type: "text",
|
|
31
|
+
},
|
|
32
|
+
],
|
|
33
|
+
},
|
|
34
|
+
metrics: [],
|
|
35
|
+
models: [
|
|
36
|
+
{
|
|
37
|
+
description: "A lightweight model that has capabilities of taking both audio and text as inputs and generating responses.",
|
|
38
|
+
id: "fixie-ai/ultravox-v0_5-llama-3_2-1b",
|
|
39
|
+
},
|
|
40
|
+
{
|
|
41
|
+
description: "A multimodal model that supports voice chat and audio analysis.",
|
|
42
|
+
id: "Qwen/Qwen2-Audio-7B-Instruct",
|
|
43
|
+
},
|
|
44
|
+
{
|
|
45
|
+
description: "A model for audio understanding, speech translation, and transcription.",
|
|
46
|
+
id: "mistralai/Voxtral-Small-24B-2507",
|
|
47
|
+
},
|
|
48
|
+
{
|
|
49
|
+
description: "A new model capable of audio question answering and reasoning.",
|
|
50
|
+
id: "nvidia/audio-flamingo-3",
|
|
51
|
+
},
|
|
52
|
+
],
|
|
53
|
+
spaces: [
|
|
54
|
+
{
|
|
55
|
+
description: "A space that takes input as both audio and text and generates answers.",
|
|
56
|
+
id: "iamomtiwari/ATTT",
|
|
57
|
+
},
|
|
58
|
+
{
|
|
59
|
+
description: "A web application that demonstrates chatting with the Qwen2Audio Model.",
|
|
60
|
+
id: "freddyaboulton/talk-to-qwen-webrtc",
|
|
61
|
+
},
|
|
62
|
+
],
|
|
63
|
+
summary: "Audio-text-to-text models take both an audio clip and a text prompt as input, and generate natural language text as output. These models can answer questions about spoken content, summarize meetings, analyze music, or interpret speech beyond simple transcription. They are useful for applications that combine speech understanding with reasoning or conversation.",
|
|
64
|
+
widgetModels: [],
|
|
65
|
+
youtubeId: "",
|
|
66
|
+
};
|
|
67
|
+
exports.default = taskData;
|
package/dist/esm/local-apps.d.ts
CHANGED
|
@@ -207,7 +207,7 @@ export declare const LOCAL_APPS: {
|
|
|
207
207
|
prettyLabel: string;
|
|
208
208
|
docsUrl: string;
|
|
209
209
|
mainTask: "text-generation";
|
|
210
|
-
displayOnModelPage:
|
|
210
|
+
displayOnModelPage: (model: ModelData) => boolean;
|
|
211
211
|
snippet: (model: ModelData, filepath?: string) => LocalAppSnippet[];
|
|
212
212
|
};
|
|
213
213
|
};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"local-apps.d.ts","sourceRoot":"","sources":["../../src/local-apps.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAKnD,MAAM,WAAW,eAAe;IAC/B;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;CAC3B;AAED;;GAEG;AACH,MAAM,MAAM,QAAQ,GAAG;IACtB;;OAEG;IACH,WAAW,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,QAAQ,EAAE,YAAY,CAAC;IACvB;;OAEG;IACH,SAAS,CAAC,EAAE,OAAO,CAAC;IAEpB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB;;OAEG;IACH,kBAAkB,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,OAAO,CAAC;CAClD,GAAG,CACD;IACA;;OAEG;IACH,QAAQ,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,MAAM,KAAK,GAAG,CAAC;CACtD,GACD;IACA;;;;OAIG;IACH,OAAO,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,MAAM,KAAK,MAAM,GAAG,MAAM,EAAE,GAAG,eAAe,GAAG,eAAe,EAAE,CAAC;CACzG,CACH,CAAC;AAqBF,iBAAS,UAAU,CAAC,KAAK,EAAE,SAAS,GAAG,OAAO,CAE7C;AAED,iBAAS,mBAAmB,CAAC,KAAK,EAAE,SAAS,WAE5C;
|
|
1
|
+
{"version":3,"file":"local-apps.d.ts","sourceRoot":"","sources":["../../src/local-apps.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAKnD,MAAM,WAAW,eAAe;IAC/B;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;CAC3B;AAED;;GAEG;AACH,MAAM,MAAM,QAAQ,GAAG;IACtB;;OAEG;IACH,WAAW,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,QAAQ,EAAE,YAAY,CAAC;IACvB;;OAEG;IACH,SAAS,CAAC,EAAE,OAAO,CAAC;IAEpB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB;;OAEG;IACH,kBAAkB,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,OAAO,CAAC;CAClD,GAAG,CACD;IACA;;OAEG;IACH,QAAQ,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,MAAM,KAAK,GAAG,CAAC;CACtD,GACD;IACA;;;;OAIG;IACH,OAAO,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,MAAM,KAAK,MAAM,GAAG,MAAM,EAAE,GAAG,eAAe,GAAG,eAAe,EAAE,CAAC;CACzG,CACH,CAAC;AAqBF,iBAAS,UAAU,CAAC,KAAK,EAAE,SAAS,GAAG,OAAO,CAE7C;AAED,iBAAS,mBAAmB,CAAC,KAAK,EAAE,SAAS,WAE5C;AAiRD;;;;;;;;;;GAUG;AACH,eAAO,MAAM,UAAU;;;;;;yBAvQS,SAAS,aAAa,MAAM,KAAG,eAAe,EAAE;;;;;;;yBAsCzC,SAAS,aAAa,MAAM,KAAG,eAAe,EAAE;;;;;;oCAoPzD,SAAS;yBA3MX,SAAS,KAAG,eAAe,EAAE;;;;;;;yBA4F5B,SAAS,KAAG,eAAe,EAAE;;;;;;;yBA7B/B,SAAS,KAAG,eAAe,EAAE;;;;;;;;;;;;;;yBAtFzB,SAAS,aAAa,MAAM,KAAG,eAAe,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;yBAJjD,SAAS,aAAa,MAAM,KAAG,MAAM;;;;;;;yBA0J1B,SAAS,aAAa,MAAM,KAAG,MAAM;;;;;;;yBAI9C,SAAS,aAAa,MAAM,KAAG,eAAe,EAAE;;CAkO5C,CAAC;AAErC,MAAM,MAAM,WAAW,GAAG,MAAM,OAAO,UAAU,CAAC"}
|
package/dist/esm/local-apps.js
CHANGED
|
@@ -22,6 +22,9 @@ function isTgiModel(model) {
|
|
|
22
22
|
function isLlamaCppGgufModel(model) {
|
|
23
23
|
return !!model.gguf?.context_length;
|
|
24
24
|
}
|
|
25
|
+
function isAmdRyzenModel(model) {
|
|
26
|
+
return model.tags.includes("ryzenai-hybrid") || model.tags.includes("ryzenai-npu");
|
|
27
|
+
}
|
|
25
28
|
function isMlxModel(model) {
|
|
26
29
|
return model.tags.includes("mlx");
|
|
27
30
|
}
|
|
@@ -232,21 +235,37 @@ const snippetDockerModelRunner = (model, filepath) => {
|
|
|
232
235
|
};
|
|
233
236
|
const snippetLemonade = (model, filepath) => {
|
|
234
237
|
const tagName = getQuantTag(filepath);
|
|
235
|
-
const modelName = model.id.split("/")[1];
|
|
238
|
+
const modelName = model.id.includes("/") ? model.id.split("/")[1] : model.id;
|
|
239
|
+
// Get recipe according to model type
|
|
240
|
+
let simplifiedModelName;
|
|
241
|
+
let recipe;
|
|
242
|
+
let checkpoint;
|
|
243
|
+
let requirements;
|
|
244
|
+
if (model.tags.some((tag) => ["ryzenai-npu", "ryzenai-hybrid"].includes(tag))) {
|
|
245
|
+
recipe = model.tags.includes("ryzenai-npu") ? "oga-npu" : "oga-hybrid";
|
|
246
|
+
checkpoint = model.id;
|
|
247
|
+
requirements = " (requires RyzenAI 300 series)";
|
|
248
|
+
simplifiedModelName = modelName.split("-awq-")[0];
|
|
249
|
+
simplifiedModelName += recipe === "oga-npu" ? "-NPU" : "-Hybrid";
|
|
250
|
+
}
|
|
251
|
+
else {
|
|
252
|
+
recipe = "llamacpp";
|
|
253
|
+
checkpoint = `${model.id}${tagName}`;
|
|
254
|
+
requirements = "";
|
|
255
|
+
simplifiedModelName = modelName;
|
|
256
|
+
}
|
|
236
257
|
return [
|
|
237
258
|
{
|
|
238
259
|
title: "Pull the model",
|
|
239
260
|
setup: "# Download Lemonade from https://lemonade-server.ai/",
|
|
240
261
|
content: [
|
|
241
|
-
`lemonade-server pull user.${
|
|
242
|
-
--checkpoint ${model.id}${tagName} \\
|
|
243
|
-
--recipe llamacpp`,
|
|
262
|
+
`lemonade-server pull user.${simplifiedModelName} --checkpoint ${checkpoint} --recipe ${recipe}`,
|
|
244
263
|
"# Note: If you installed from source, use the lemonade-server-dev command instead.",
|
|
245
264
|
].join("\n"),
|
|
246
265
|
},
|
|
247
266
|
{
|
|
248
|
-
title:
|
|
249
|
-
content: `lemonade-server run user.${
|
|
267
|
+
title: `Run and chat with the model${requirements}`,
|
|
268
|
+
content: `lemonade-server run user.${simplifiedModelName}`,
|
|
250
269
|
},
|
|
251
270
|
{
|
|
252
271
|
title: "List all available models",
|
|
@@ -433,7 +452,7 @@ export const LOCAL_APPS = {
|
|
|
433
452
|
prettyLabel: "Lemonade",
|
|
434
453
|
docsUrl: "https://lemonade-server.ai",
|
|
435
454
|
mainTask: "text-generation",
|
|
436
|
-
displayOnModelPage: isLlamaCppGgufModel,
|
|
455
|
+
displayOnModelPage: (model) => isLlamaCppGgufModel(model) || isAmdRyzenModel(model),
|
|
437
456
|
snippet: snippetLemonade,
|
|
438
457
|
},
|
|
439
458
|
};
|
package/dist/esm/model-data.d.ts
CHANGED
|
@@ -43,6 +43,10 @@ export interface ModelData {
|
|
|
43
43
|
quant_method?: string;
|
|
44
44
|
};
|
|
45
45
|
tokenizer_config?: TokenizerConfig;
|
|
46
|
+
processor_config?: {
|
|
47
|
+
chat_template?: string;
|
|
48
|
+
};
|
|
49
|
+
chat_template_jinja?: string;
|
|
46
50
|
adapter_transformers?: {
|
|
47
51
|
model_name?: string;
|
|
48
52
|
model_class?: string;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"model-data.d.ts","sourceRoot":"","sources":["../../src/model-data.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACzD,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAE3D;;GAEG;AACH,MAAM,WAAW,SAAS;IACzB;;OAEG;IACH,EAAE,EAAE,MAAM,CAAC;IACX;;;OAGG;IACH,SAAS,EAAE,MAAM,CAAC;IAClB;;OAEG;IACH,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB;;OAEG;IACH,MAAM,CAAC,EAAE;QACR,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;QACzB;;WAEG;QACH,QAAQ,CAAC,EAAE;YACV;;eAEG;YACH,CAAC,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;SACpB,CAAC;QACF,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,mBAAmB,CAAC,EAAE;YACrB,IAAI,CAAC,EAAE,MAAM,CAAC;YACd,YAAY,CAAC,EAAE,OAAO,CAAC;YACvB,YAAY,CAAC,EAAE,OAAO,CAAC;YACvB;;eAEG;YACH,YAAY,CAAC,EAAE,MAAM,CAAC;SACtB,CAAC;QACF,gBAAgB,CAAC,EAAE,eAAe,CAAC;QACnC,oBAAoB,CAAC,EAAE;YACtB,UAAU,CAAC,EAAE,MAAM,CAAC;YACpB,WAAW,CAAC,EAAE,MAAM,CAAC;SACrB,CAAC;QACF,SAAS,CAAC,EAAE;YACX,WAAW,CAAC,EAAE,MAAM,CAAC;SACrB,CAAC;QACF,OAAO,CAAC,EAAE;YACT,KAAK,CAAC,EAAE;gBACP,IAAI,CAAC,EAAE,MAAM,CAAC;aACd,CAAC;YACF,YAAY,CAAC,EAAE,MAAM,CAAC;SACtB,CAAC;QACF,WAAW,CAAC,EAAE;YACb,qBAAqB,CAAC,EAAE,MAAM,CAAC;YAC/B,iBAAiB,CAAC,EAAE,MAAM,CAAC;YAC3B,gBAAgB,CAAC,EAAE,MAAM,CAAC;SAC1B,CAAC;QACF,IAAI,CAAC,EAAE;YACN,uBAAuB,CAAC,EAAE,MAAM,CAAC;YACjC,SAAS,CAAC,EAAE,MAAM,CAAC;SACnB,CAAC;QACF,SAAS,CAAC,EAAE;YACX,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;SACjB,CAAC;KACF,CAAC;IACF;;OAEG;IACH,IAAI,EAAE,MAAM,EAAE,CAAC;IACf;;OAEG;IACH,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;IACpC;;OAEG;IACH,YAAY,CAAC,EAAE,YAAY,GAAG,SAAS,CAAC;IACxC;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAChC;;;;;OAKG;IACH,UAAU,CAAC,EAAE,aAAa,EAAE,GAAG,SAAS,CAAC;IACzC;;;;;;;;;OASG;IACH,QAAQ,CAAC,EAAE;QACV,SAAS,CAAC,EACP,OAAO,GACP;YACA,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;SACpC,CAAC;QACL,UAAU,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;QAC/B,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;KAChC,CAAC;IACF;;;OAGG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,WAAW,CAAC,EAAE;QACb,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACnC,KAAK,EAAE,MAAM,CAAC;QACd,OAAO,EAAE,OAAO,CAAC;KACjB,CAAC;IACF,IAAI,CAAC,EAAE;QACN,KAAK,EAAE,MAAM,CAAC;QACd,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,cAAc,CAAC,EAAE,MAAM,CAAC;KACxB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;OAEG;IACH,UAAU,EAAE,MAAM,CAAC;IACnB;;OAEG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB;;OAEG;IACH,YAAY,CAAC,EAAE,YAAY,CAAC;IAC5B;;OAEG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;CACnB"}
|
|
1
|
+
{"version":3,"file":"model-data.d.ts","sourceRoot":"","sources":["../../src/model-data.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACzD,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAE3D;;GAEG;AACH,MAAM,WAAW,SAAS;IACzB;;OAEG;IACH,EAAE,EAAE,MAAM,CAAC;IACX;;;OAGG;IACH,SAAS,EAAE,MAAM,CAAC;IAClB;;OAEG;IACH,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB;;OAEG;IACH,MAAM,CAAC,EAAE;QACR,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;QACzB;;WAEG;QACH,QAAQ,CAAC,EAAE;YACV;;eAEG;YACH,CAAC,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;SACpB,CAAC;QACF,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,mBAAmB,CAAC,EAAE;YACrB,IAAI,CAAC,EAAE,MAAM,CAAC;YACd,YAAY,CAAC,EAAE,OAAO,CAAC;YACvB,YAAY,CAAC,EAAE,OAAO,CAAC;YACvB;;eAEG;YACH,YAAY,CAAC,EAAE,MAAM,CAAC;SACtB,CAAC;QACF,gBAAgB,CAAC,EAAE,eAAe,CAAC;QACnC,gBAAgB,CAAC,EAAE;YAClB,aAAa,CAAC,EAAE,MAAM,CAAC;SACvB,CAAC;QACF,mBAAmB,CAAC,EAAE,MAAM,CAAC;QAC7B,oBAAoB,CAAC,EAAE;YACtB,UAAU,CAAC,EAAE,MAAM,CAAC;YACpB,WAAW,CAAC,EAAE,MAAM,CAAC;SACrB,CAAC;QACF,SAAS,CAAC,EAAE;YACX,WAAW,CAAC,EAAE,MAAM,CAAC;SACrB,CAAC;QACF,OAAO,CAAC,EAAE;YACT,KAAK,CAAC,EAAE;gBACP,IAAI,CAAC,EAAE,MAAM,CAAC;aACd,CAAC;YACF,YAAY,CAAC,EAAE,MAAM,CAAC;SACtB,CAAC;QACF,WAAW,CAAC,EAAE;YACb,qBAAqB,CAAC,EAAE,MAAM,CAAC;YAC/B,iBAAiB,CAAC,EAAE,MAAM,CAAC;YAC3B,gBAAgB,CAAC,EAAE,MAAM,CAAC;SAC1B,CAAC;QACF,IAAI,CAAC,EAAE;YACN,uBAAuB,CAAC,EAAE,MAAM,CAAC;YACjC,SAAS,CAAC,EAAE,MAAM,CAAC;SACnB,CAAC;QACF,SAAS,CAAC,EAAE;YACX,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;SACjB,CAAC;KACF,CAAC;IACF;;OAEG;IACH,IAAI,EAAE,MAAM,EAAE,CAAC;IACf;;OAEG;IACH,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;IACpC;;OAEG;IACH,YAAY,CAAC,EAAE,YAAY,GAAG,SAAS,CAAC;IACxC;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAChC;;;;;OAKG;IACH,UAAU,CAAC,EAAE,aAAa,EAAE,GAAG,SAAS,CAAC;IACzC;;;;;;;;;OASG;IACH,QAAQ,CAAC,EAAE;QACV,SAAS,CAAC,EACP,OAAO,GACP;YACA,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;SACpC,CAAC;QACL,UAAU,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;QAC/B,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;KAChC,CAAC;IACF;;;OAGG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,WAAW,CAAC,EAAE;QACb,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACnC,KAAK,EAAE,MAAM,CAAC;QACd,OAAO,EAAE,OAAO,CAAC;KACjB,CAAC;IACF,IAAI,CAAC,EAAE;QACN,KAAK,EAAE,MAAM,CAAC;QACd,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,cAAc,CAAC,EAAE,MAAM,CAAC;KACxB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;OAEG;IACH,UAAU,EAAE,MAAM,CAAC;IACnB;;OAEG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB;;OAEG;IACH,YAAY,CAAC,EAAE,YAAY,CAAC;IAC5B;;OAEG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;CACnB"}
|
|
@@ -86,6 +86,7 @@ export declare const univa: (model: ModelData) => string[];
|
|
|
86
86
|
export declare const mlxim: (model: ModelData) => string[];
|
|
87
87
|
export declare const mlx: (model: ModelData) => string[];
|
|
88
88
|
export declare const model2vec: (model: ModelData) => string[];
|
|
89
|
+
export declare const pruna: (model: ModelData) => string[];
|
|
89
90
|
export declare const nemo: (model: ModelData) => string[];
|
|
90
91
|
export declare const outetts: (model: ModelData) => string[];
|
|
91
92
|
export declare const pxia: (model: ModelData) => string[];
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"model-libraries-snippets.d.ts","sourceRoot":"","sources":["../../src/model-libraries-snippets.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAkBjD,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAKjD,CAAC;AAkBF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAKjD,CAAC;AAEF,eAAO,MAAM,OAAO,GAAI,OAAO,SAAS,KAAG,MAAM,EAIhD,CAAC;AAEF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAkBlD,CAAC;AAaF,eAAO,MAAM,IAAI,GAAI,OAAO,SAAS,KAAG,MAAM,EAY7C,CAAC;AAEF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAenC,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAsDnC,CAAC;AAEF,eAAO,MAAM,cAAc,QAAO,MAAM,EAcvC,CAAC;AAEF,eAAO,MAAM,iBAAiB,GAAI,OAAO,SAAS,KAAG,MAAM,EA6C1D,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAwBlD,CAAC;AAEF,eAAO,MAAM,eAAe,QAAO,MAAM,EAoBxC,CAAC;AAEF,eAAO,MAAM,GAAG,GAAI,OAAO,SAAS,KAAG,MAAM,EAS5C,CAAC;AAEF,eAAO,MAAM,iBAAiB,GAAI,OAAO,SAAS,KAAG,MAAM,EAY1D,CAAC;AAkKF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EA6BlD,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,OAAO,SAAS,KAAG,MAAM,EAwCrD,CAAC;AAEF,eAAO,MAAM,gBAAgB,GAAI,OAAO,SAAS,KAAG,MAAM,EAgBzD,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,OAAO,SAAS,KAAG,MAAM,EAmBrD,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAgB/C,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAMlD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EASlD,CAAC;AAIF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAO/C,CAAC;AAEF,eAAO,MAAM,OAAO,GAAI,OAAO,SAAS,KAAG,MAAM,EAMhD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAgBjD,CAAC;AAEF,eAAO,MAAM,OAAO,GAAI,OAAO,SAAS,KAAG,MAAM,EAehD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAS9C,CAAC;AA4EF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAsBlD,CAAC;AAEF,eAAO,MAAM,UAAU,GAAI,OAAO,SAAS,KAAG,MAAM,EA+BnD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EASlD,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,OAAO,SAAS,KAAG,MAAM,EA+BrD,CAAC;AAEF,eAAO,MAAM,gBAAgB,GAAI,OAAO,SAAS,KAAG,MAAM,EA4BzD,CAAC;AAEF,eAAO,MAAM,OAAO,GAAI,OAAO,SAAS,KAAG,MAAM,EAuChD,CAAC;AAEF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAOjD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAIlD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAKlD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAQlD,CAAC;AAEF,eAAO,MAAM,aAAa,QAAO,MAAM,EAQtC,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAKlD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAsBlD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAmDlD,CAAC;AAEF,eAAO,MAAM,kBAAkB,GAAI,OAAO,SAAS,KAAG,MAAM,EAgB3D,CAAC;AACF,eAAO,MAAM,WAAW,GAAI,OAAO,SAAS,KAAG,MAAM,EAUpD,CAAC;AAEF,eAAO,MAAM,uBAAuB,GAAI,OAAO,SAAS,KAAG,MAAM,EAehE,CAAC;AAiBF,eAAO,MAAM,cAAc,GAAI,OAAO,SAAS,KAAG,MAAM,EAKvD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,OAAO,SAAS,KAAG,MAAM,EAKrD,CAAC;AAyBF,eAAO,MAAM,aAAa,GAAI,OAAO,SAAS,KAAG,MAAM,EAOtD,CAAC;AAEF,eAAO,MAAM,IAAI,GAAI,OAAO,SAAS,KAAG,MAAM,EAI7C,CAAC;AAEF,eAAO,MAAM,OAAO,QAA6B,MAAM,EAQtD,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAanC,CAAC;AAsCF,eAAO,MAAM,OAAO,GAAI,OAAO,SAAS,KAAG,MAAM,EAehD,CAAC;AAEF,eAAO,MAAM,kBAAkB,GAAI,OAAO,SAAS,KAAG,MAAM,EAmC3D,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,IAAI,GAAI,OAAO,SAAS,KAAG,MAAM,EA2B7C,CAAC;AAEF,eAAO,MAAM,aAAa,GAAI,OAAO,SAAS,KAAG,MAAM,EAEtD,CAAC;AASF,eAAO,MAAM,oBAAoB,GAAI,OAAO,SAAS,KAAG,MAAM,EAuC7D,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAU9C,CAAC;AAEF,eAAO,MAAM,WAAW,GAAI,OAAO,SAAS,KAAG,MAAM,EAIpD,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAK/C,CAAC;AAkBF,eAAO,MAAM,WAAW,GAAI,OAAO,SAAS,KAAG,MAAM,EAkBpD,CAAC;AAEF,eAAO,MAAM,UAAU,GAAI,OAAO,SAAS,KAAG,MAAM,EAInD,CAAC;
|
|
1
|
+
{"version":3,"file":"model-libraries-snippets.d.ts","sourceRoot":"","sources":["../../src/model-libraries-snippets.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAkBjD,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAKjD,CAAC;AAkBF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAKjD,CAAC;AAEF,eAAO,MAAM,OAAO,GAAI,OAAO,SAAS,KAAG,MAAM,EAIhD,CAAC;AAEF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAkBlD,CAAC;AAaF,eAAO,MAAM,IAAI,GAAI,OAAO,SAAS,KAAG,MAAM,EAY7C,CAAC;AAEF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAenC,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAsDnC,CAAC;AAEF,eAAO,MAAM,cAAc,QAAO,MAAM,EAcvC,CAAC;AAEF,eAAO,MAAM,iBAAiB,GAAI,OAAO,SAAS,KAAG,MAAM,EA6C1D,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAwBlD,CAAC;AAEF,eAAO,MAAM,eAAe,QAAO,MAAM,EAoBxC,CAAC;AAEF,eAAO,MAAM,GAAG,GAAI,OAAO,SAAS,KAAG,MAAM,EAS5C,CAAC;AAEF,eAAO,MAAM,iBAAiB,GAAI,OAAO,SAAS,KAAG,MAAM,EAY1D,CAAC;AAkKF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EA6BlD,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,OAAO,SAAS,KAAG,MAAM,EAwCrD,CAAC;AAEF,eAAO,MAAM,gBAAgB,GAAI,OAAO,SAAS,KAAG,MAAM,EAgBzD,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,OAAO,SAAS,KAAG,MAAM,EAmBrD,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAgB/C,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAMlD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EASlD,CAAC;AAIF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAO/C,CAAC;AAEF,eAAO,MAAM,OAAO,GAAI,OAAO,SAAS,KAAG,MAAM,EAMhD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAgBjD,CAAC;AAEF,eAAO,MAAM,OAAO,GAAI,OAAO,SAAS,KAAG,MAAM,EAehD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAS9C,CAAC;AA4EF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAsBlD,CAAC;AAEF,eAAO,MAAM,UAAU,GAAI,OAAO,SAAS,KAAG,MAAM,EA+BnD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EASlD,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,OAAO,SAAS,KAAG,MAAM,EA+BrD,CAAC;AAEF,eAAO,MAAM,gBAAgB,GAAI,OAAO,SAAS,KAAG,MAAM,EA4BzD,CAAC;AAEF,eAAO,MAAM,OAAO,GAAI,OAAO,SAAS,KAAG,MAAM,EAuChD,CAAC;AAEF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAOjD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAIlD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAKlD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAQlD,CAAC;AAEF,eAAO,MAAM,aAAa,QAAO,MAAM,EAQtC,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAKlD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAsBlD,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAmDlD,CAAC;AAEF,eAAO,MAAM,kBAAkB,GAAI,OAAO,SAAS,KAAG,MAAM,EAgB3D,CAAC;AACF,eAAO,MAAM,WAAW,GAAI,OAAO,SAAS,KAAG,MAAM,EAUpD,CAAC;AAEF,eAAO,MAAM,uBAAuB,GAAI,OAAO,SAAS,KAAG,MAAM,EAehE,CAAC;AAiBF,eAAO,MAAM,cAAc,GAAI,OAAO,SAAS,KAAG,MAAM,EAKvD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,OAAO,SAAS,KAAG,MAAM,EAKrD,CAAC;AAyBF,eAAO,MAAM,aAAa,GAAI,OAAO,SAAS,KAAG,MAAM,EAOtD,CAAC;AAEF,eAAO,MAAM,IAAI,GAAI,OAAO,SAAS,KAAG,MAAM,EAI7C,CAAC;AAEF,eAAO,MAAM,OAAO,QAA6B,MAAM,EAQtD,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAanC,CAAC;AAsCF,eAAO,MAAM,OAAO,GAAI,OAAO,SAAS,KAAG,MAAM,EAehD,CAAC;AAEF,eAAO,MAAM,kBAAkB,GAAI,OAAO,SAAS,KAAG,MAAM,EAmC3D,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,IAAI,GAAI,OAAO,SAAS,KAAG,MAAM,EA2B7C,CAAC;AAEF,eAAO,MAAM,aAAa,GAAI,OAAO,SAAS,KAAG,MAAM,EAEtD,CAAC;AASF,eAAO,MAAM,oBAAoB,GAAI,OAAO,SAAS,KAAG,MAAM,EAuC7D,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAU9C,CAAC;AAEF,eAAO,MAAM,WAAW,GAAI,OAAO,SAAS,KAAG,MAAM,EAIpD,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAK/C,CAAC;AAkBF,eAAO,MAAM,WAAW,GAAI,OAAO,SAAS,KAAG,MAAM,EAkBpD,CAAC;AAEF,eAAO,MAAM,UAAU,GAAI,OAAO,SAAS,KAAG,MAAM,EAInD,CAAC;AAOF,eAAO,MAAM,YAAY,GAAI,OAAO,SAAS,KAAG,MAAM,EAyGrD,CAAC;AAEF,eAAO,MAAM,cAAc,GAAI,OAAO,SAAS,KAAG,MAAM,EAcvD,CAAC;AAiBF,eAAO,MAAM,IAAI,GAAI,OAAO,SAAS,KAAG,MAAM,EAiB7C,CAAC;AAEF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAKjD,CAAC;AAEF,eAAO,MAAM,gBAAgB,GAAI,OAAO,SAAS,KAAG,MAAM,EAMzD,CAAC;AAgBF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAEjD,CAAC;AAEF,eAAO,MAAM,MAAM,QAA6B,MAAM,EAMrD,CAAC;AAEF,eAAO,MAAM,IAAI,GAAI,OAAO,SAAS,KAAG,MAAM,EAkB7C,CAAC;AAEF,eAAO,MAAM,UAAU,GAAI,OAAO,SAAS,KAAG,MAAM,EAWnD,CAAC;AAEF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAQ/C,CAAC;AAEF,eAAO,MAAM,UAAU,GAAI,OAAO,SAAS,KAAG,MAAM,EAInD,CAAC;AAEF,eAAO,MAAM,GAAG,QAAO,MAAM,EAgB5B,CAAC;AAEF,eAAO,MAAM,OAAO,QAAO,MAAM,EAYhC,CAAC;AAEF,eAAO,MAAM,WAAW,GAAI,OAAO,SAAS,KAAG,MAAM,EAiBpD,CAAC;AAEF,eAAO,MAAM,QAAQ,GAAI,OAAO,SAAS,KAAG,MAAM,EAYjD,CAAC;AAEF,eAAO,MAAM,WAAW,GAAI,OAAO,SAAS,KAAG,MAAM,EAKpD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAW9C,CAAC;AAmEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,GAAG,GAAI,OAAO,SAAS,KAAG,MAAM,EAY5C,CAAC;AAEF,eAAO,MAAM,SAAS,GAAI,OAAO,SAAS,KAAG,MAAM,EAIlD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EA0B9C,CAAC;AAuDF,eAAO,MAAM,IAAI,GAAI,OAAO,SAAS,KAAG,MAAM,EAQ7C,CAAC;AAEF,eAAO,MAAM,OAAO,GAAI,OAAO,SAAS,KAAG,MAAM,EAuBhD,CAAC;AAEF,eAAO,MAAM,IAAI,GAAI,OAAO,SAAS,KAAG,MAAM,EAI7C,CAAC;AAEF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAI/C,CAAC;AA4BF,eAAO,MAAM,MAAM,GAAI,OAAO,SAAS,KAAG,MAAM,EAO/C,CAAC;AAEF,eAAO,MAAM,UAAU,GAAI,OAAO,SAAS,KAAG,MAAM,EAUnD,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAYnC,CAAC;AAEF,eAAO,MAAM,cAAc,GAAI,OAAO,SAAS,KAAG,MAAM,EAKvD,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,KAAK,GAAI,OAAO,SAAS,KAAG,MAAM,EAiB9C,CAAC"}
|
|
@@ -1357,6 +1357,9 @@ export const terratorch = (model) => [
|
|
|
1357
1357
|
|
|
1358
1358
|
model = BACKBONE_REGISTRY.build("${model.id}")`,
|
|
1359
1359
|
];
|
|
1360
|
+
const hasChatTemplate = (model) => model.config?.tokenizer_config?.chat_template !== undefined ||
|
|
1361
|
+
model.config?.processor_config?.chat_template !== undefined ||
|
|
1362
|
+
model.config?.chat_template_jinja !== undefined;
|
|
1360
1363
|
export const transformers = (model) => {
|
|
1361
1364
|
const info = model.transformersInfo;
|
|
1362
1365
|
if (!info) {
|
|
@@ -1371,7 +1374,7 @@ export const transformers = (model) => {
|
|
|
1371
1374
|
? "extractor"
|
|
1372
1375
|
: "processor";
|
|
1373
1376
|
autoSnippet.push("# Load model directly", `from transformers import ${info.processor}, ${info.auto_model}`, "", `${processorVarName} = ${info.processor}.from_pretrained("${model.id}"` + remote_code_snippet + ")", `model = ${info.auto_model}.from_pretrained("${model.id}"` + remote_code_snippet + ")");
|
|
1374
|
-
if (model.tags.includes("conversational")) {
|
|
1377
|
+
if (model.tags.includes("conversational") && hasChatTemplate(model)) {
|
|
1375
1378
|
if (model.tags.includes("image-text-to-text")) {
|
|
1376
1379
|
autoSnippet.push("messages = [", [
|
|
1377
1380
|
" {",
|
|
@@ -1719,6 +1722,67 @@ export const model2vec = (model) => [
|
|
|
1719
1722
|
|
|
1720
1723
|
model = StaticModel.from_pretrained("${model.id}")`,
|
|
1721
1724
|
];
|
|
1725
|
+
export const pruna = (model) => {
|
|
1726
|
+
let snippets;
|
|
1727
|
+
if (model.tags.includes("diffusers")) {
|
|
1728
|
+
snippets = pruna_diffusers(model);
|
|
1729
|
+
}
|
|
1730
|
+
else if (model.tags.includes("transformers")) {
|
|
1731
|
+
snippets = pruna_transformers(model);
|
|
1732
|
+
}
|
|
1733
|
+
else {
|
|
1734
|
+
snippets = pruna_default(model);
|
|
1735
|
+
}
|
|
1736
|
+
const ensurePrunaModelImport = (snippet) => {
|
|
1737
|
+
if (!/^from pruna import PrunaModel/m.test(snippet)) {
|
|
1738
|
+
return `from pruna import PrunaModel\n${snippet}`;
|
|
1739
|
+
}
|
|
1740
|
+
return snippet;
|
|
1741
|
+
};
|
|
1742
|
+
snippets = snippets.map(ensurePrunaModelImport);
|
|
1743
|
+
if (model.tags.includes("pruna_pro-ai")) {
|
|
1744
|
+
return snippets.map((snippet) => snippet.replace(/\bpruna\b/g, "pruna_pro").replace(/\bPrunaModel\b/g, "PrunaProModel"));
|
|
1745
|
+
}
|
|
1746
|
+
return snippets;
|
|
1747
|
+
};
|
|
1748
|
+
const pruna_diffusers = (model) => {
|
|
1749
|
+
const diffusersSnippets = diffusers(model);
|
|
1750
|
+
return diffusersSnippets.map((snippet) => snippet
|
|
1751
|
+
// Replace pipeline classes with PrunaModel
|
|
1752
|
+
.replace(/\b\w*Pipeline\w*\b/g, "PrunaModel")
|
|
1753
|
+
// Clean up diffusers imports containing PrunaModel
|
|
1754
|
+
.replace(/from diffusers import ([^,\n]*PrunaModel[^,\n]*)/g, "")
|
|
1755
|
+
.replace(/from diffusers import ([^,\n]+),?\s*([^,\n]*PrunaModel[^,\n]*)/g, "from diffusers import $1")
|
|
1756
|
+
.replace(/from diffusers import\s*(\n|$)/g, "")
|
|
1757
|
+
// Fix PrunaModel imports
|
|
1758
|
+
.replace(/from diffusers import PrunaModel/g, "from pruna import PrunaModel")
|
|
1759
|
+
.replace(/from diffusers import ([^,\n]+), PrunaModel/g, "from diffusers import $1")
|
|
1760
|
+
.replace(/from diffusers import PrunaModel, ([^,\n]+)/g, "from diffusers import $1")
|
|
1761
|
+
// Clean up whitespace
|
|
1762
|
+
.replace(/\n\n+/g, "\n")
|
|
1763
|
+
.trim());
|
|
1764
|
+
};
|
|
1765
|
+
const pruna_transformers = (model) => {
|
|
1766
|
+
const info = model.transformersInfo;
|
|
1767
|
+
const transformersSnippets = transformers(model);
|
|
1768
|
+
// Replace pipeline with PrunaModel
|
|
1769
|
+
let processedSnippets = transformersSnippets.map((snippet) => snippet
|
|
1770
|
+
.replace(/from transformers import pipeline/g, "from pruna import PrunaModel")
|
|
1771
|
+
.replace(/pipeline\([^)]*\)/g, `PrunaModel.from_pretrained("${model.id}")`));
|
|
1772
|
+
// Additional cleanup if auto_model info is available
|
|
1773
|
+
if (info?.auto_model) {
|
|
1774
|
+
processedSnippets = processedSnippets.map((snippet) => snippet
|
|
1775
|
+
.replace(new RegExp(`from transformers import ${info.auto_model}\n?`, "g"), "")
|
|
1776
|
+
.replace(new RegExp(`${info.auto_model}.from_pretrained`, "g"), "PrunaModel.from_pretrained")
|
|
1777
|
+
.replace(new RegExp(`^.*from.*import.*(, *${info.auto_model})+.*$`, "gm"), (line) => line.replace(new RegExp(`, *${info.auto_model}`, "g"), "")));
|
|
1778
|
+
}
|
|
1779
|
+
return processedSnippets;
|
|
1780
|
+
};
|
|
1781
|
+
const pruna_default = (model) => [
|
|
1782
|
+
`from pruna import PrunaModel
|
|
1783
|
+
model = PrunaModel.from_pretrained("${model.id}")
|
|
1784
|
+
`,
|
|
1785
|
+
];
|
|
1722
1786
|
export const nemo = (model) => {
|
|
1723
1787
|
let command = undefined;
|
|
1724
1788
|
// Resolve the tag to a nemo domain/sub-domain
|
|
@@ -792,6 +792,13 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
|
|
|
792
792
|
filter: false;
|
|
793
793
|
countDownloads: string;
|
|
794
794
|
};
|
|
795
|
+
"pruna-ai": {
|
|
796
|
+
prettyLabel: string;
|
|
797
|
+
repoName: string;
|
|
798
|
+
repoUrl: string;
|
|
799
|
+
snippets: (model: ModelData) => string[];
|
|
800
|
+
docsUrl: string;
|
|
801
|
+
};
|
|
795
802
|
pxia: {
|
|
796
803
|
prettyLabel: string;
|
|
797
804
|
repoName: string;
|
|
@@ -1247,5 +1254,5 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
|
|
|
1247
1254
|
};
|
|
1248
1255
|
export type ModelLibraryKey = keyof typeof MODEL_LIBRARIES_UI_ELEMENTS;
|
|
1249
1256
|
export declare const ALL_MODEL_LIBRARY_KEYS: ModelLibraryKey[];
|
|
1250
|
-
export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("acestep" | "adapter-transformers" | "allennlp" | "anemoi" | "araclip" | "asteroid" | "audiocraft" | "audioseal" | "bagel-mot" | "bboxmaskpose" | "ben2" | "bertopic" | "big_vision" | "birder" | "birefnet" | "bm25s" | "champ" | "chatterbox" | "chat_tts" | "colpali" | "comet" | "contexttab" | "cosmos" | "cxr-foundation" | "deepforest" | "depth-anything-v2" | "depth-pro" | "derm-foundation" | "describe-anything" | "dia-tts" | "diffree" | "diffusers" | "diffusionkit" | "doctr" | "cartesia_pytorch" | "cartesia_mlx" | "clipscope" | "cosyvoice" | "cotracker" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "fme" | "gemma.cpp" | "geometry-crafter" | "gliner" | "glyph-byt5" | "grok" | "hallo" | "hermes" | "hezar" | "htrflow" | "hunyuan-dit" | "hunyuan3d-2" | "hunyuanworld-voyager" | "imstoucan" | "index-tts" | "infinitetalk" | "infinite-you" | "keras" | "tf-keras" | "keras-hub" | "kimi-audio" | "kittentts" | "kronos" | "k2" | "lightning-ir" | "litert-lm" | "lerobot" | "liveportrait" | "llama-cpp-python" | "mini-omni2" | "mindspore" | "magi-1" | "magenta-realtime" | "mamba-ssm" | "mars5-tts" | "matanyone" | "mesh-anything" | "merlin" | "medvae" | "mitie" | "ml-agents" | "mlx" | "mlx-image" | "mlc-llm" | "model2vec" | "moshi" | "mtvcraft" | "nemo" | "open-oasis" | "open_clip" | "open-sora" | "outetts" | "paddlenlp" | "PaddleOCR" | "peft" | "perception-encoder" | "phantom-wan" | "pxia" | "pyannote-audio" | "py-feat" | "pythae" | "recurrentgemma" | "relik" | "refiners" | "renderformer" | "reverb" | "rkllm" | "saelens" | "sam2" | "sample-factory" | "sapiens" | "seedvr" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "ssr-speech" | "stable-audio-tools" | "monkeyocr" | "diffusion-single-file" | "seed-story" | "soloaudio" | "songbloom" | "stable-baselines3" | "stanza" | "swarmformer" | "f5-tts" | "genmo" | "tencent-song-generation" | "tensorflowtts" | "tabpfn" | "terratorch" | "tic-clip" | "timesfm" | "timm" | "tirex" | "torchgeo" | "transformers" | "transformers.js" | "trellis" | "ultralytics" | "univa" | "uni-3dar" | "unity-sentis" | "sana" | "videoprism" | "vfi-mamba" | "lvface" | "voicecraft" | "vui" | "wan2.2" | "wham" | "whisperkit" | "yolov10" | "zonos" | "3dtopia-xl")[];
|
|
1257
|
+
export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("acestep" | "adapter-transformers" | "allennlp" | "anemoi" | "araclip" | "asteroid" | "audiocraft" | "audioseal" | "bagel-mot" | "bboxmaskpose" | "ben2" | "bertopic" | "big_vision" | "birder" | "birefnet" | "bm25s" | "champ" | "chatterbox" | "chat_tts" | "colpali" | "comet" | "contexttab" | "cosmos" | "cxr-foundation" | "deepforest" | "depth-anything-v2" | "depth-pro" | "derm-foundation" | "describe-anything" | "dia-tts" | "diffree" | "diffusers" | "diffusionkit" | "doctr" | "cartesia_pytorch" | "cartesia_mlx" | "clipscope" | "cosyvoice" | "cotracker" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "fme" | "gemma.cpp" | "geometry-crafter" | "gliner" | "glyph-byt5" | "grok" | "hallo" | "hermes" | "hezar" | "htrflow" | "hunyuan-dit" | "hunyuan3d-2" | "hunyuanworld-voyager" | "imstoucan" | "index-tts" | "infinitetalk" | "infinite-you" | "keras" | "tf-keras" | "keras-hub" | "kimi-audio" | "kittentts" | "kronos" | "k2" | "lightning-ir" | "litert-lm" | "lerobot" | "liveportrait" | "llama-cpp-python" | "mini-omni2" | "mindspore" | "magi-1" | "magenta-realtime" | "mamba-ssm" | "mars5-tts" | "matanyone" | "mesh-anything" | "merlin" | "medvae" | "mitie" | "ml-agents" | "mlx" | "mlx-image" | "mlc-llm" | "model2vec" | "moshi" | "mtvcraft" | "nemo" | "open-oasis" | "open_clip" | "open-sora" | "outetts" | "paddlenlp" | "PaddleOCR" | "peft" | "perception-encoder" | "phantom-wan" | "pruna-ai" | "pxia" | "pyannote-audio" | "py-feat" | "pythae" | "recurrentgemma" | "relik" | "refiners" | "renderformer" | "reverb" | "rkllm" | "saelens" | "sam2" | "sample-factory" | "sapiens" | "seedvr" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "ssr-speech" | "stable-audio-tools" | "monkeyocr" | "diffusion-single-file" | "seed-story" | "soloaudio" | "songbloom" | "stable-baselines3" | "stanza" | "swarmformer" | "f5-tts" | "genmo" | "tencent-song-generation" | "tensorflowtts" | "tabpfn" | "terratorch" | "tic-clip" | "timesfm" | "timm" | "tirex" | "torchgeo" | "transformers" | "transformers.js" | "trellis" | "ultralytics" | "univa" | "uni-3dar" | "unity-sentis" | "sana" | "videoprism" | "vfi-mamba" | "lvface" | "voicecraft" | "vui" | "wan2.2" | "wham" | "whisperkit" | "yolov10" | "zonos" | "3dtopia-xl")[];
|
|
1251
1258
|
//# sourceMappingURL=model-libraries.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"model-libraries.d.ts","sourceRoot":"","sources":["../../src/model-libraries.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AAEzE;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;;;OAIG;IACH,WAAW,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,MAAM,EAAE,CAAC;IAC1C;;;;;OAKG;IACH,cAAc,CAAC,EAAE,kBAAkB,CAAC;IACpC;;;OAGG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;CACjB;AAED;;;;;;;;;;;;;GAaG;AAEH,eAAO,MAAM,2BAA2B
|
|
1
|
+
{"version":3,"file":"model-libraries.d.ts","sourceRoot":"","sources":["../../src/model-libraries.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AAEzE;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;;;OAIG;IACH,WAAW,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,MAAM,EAAE,CAAC;IAC1C;;;;;OAKG;IACH,cAAc,CAAC,EAAE,kBAAkB,CAAC;IACpC;;;OAGG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;CACjB;AAED;;;;;;;;;;;;;GAaG;AAEH,eAAO,MAAM,2BAA2B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAorCI,CAAC;AAE7C,MAAM,MAAM,eAAe,GAAG,MAAM,OAAO,2BAA2B,CAAC;AAEvE,eAAO,MAAM,sBAAsB,EAA+C,eAAe,EAAE,CAAC;AAEpG,eAAO,MAAM,8BAA8B,ivEAQ1B,CAAC"}
|
|
@@ -757,6 +757,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
|
|
|
757
757
|
filter: false,
|
|
758
758
|
countDownloads: `path_extension:"pth"`,
|
|
759
759
|
},
|
|
760
|
+
"pruna-ai": {
|
|
761
|
+
prettyLabel: "Pruna AI",
|
|
762
|
+
repoName: "Pruna AI",
|
|
763
|
+
repoUrl: "https://github.com/PrunaAI/pruna",
|
|
764
|
+
snippets: snippets.pruna,
|
|
765
|
+
docsUrl: "https://docs.pruna.ai",
|
|
766
|
+
},
|
|
760
767
|
pxia: {
|
|
761
768
|
prettyLabel: "pxia",
|
|
762
769
|
repoName: "pxia",
|
|
@@ -1028,7 +1035,7 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
|
|
|
1028
1035
|
repoUrl: "https://github.com/IBM/terratorch",
|
|
1029
1036
|
docsUrl: "https://ibm.github.io/terratorch/",
|
|
1030
1037
|
filter: false,
|
|
1031
|
-
countDownloads: `path_extension:"pt"`,
|
|
1038
|
+
countDownloads: `path_extension:"pt" OR path_extension:"ckpt"`,
|
|
1032
1039
|
snippets: snippets.terratorch,
|
|
1033
1040
|
},
|
|
1034
1041
|
"tic-clip": {
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"data.d.ts","sourceRoot":"","sources":["../../../../src/tasks/audio-text-to-text/data.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAElD,QAAA,MAAM,QAAQ,EAAE,cAiEf,CAAC;AAEF,eAAe,QAAQ,CAAC"}
|
|
@@ -0,0 +1,65 @@
|
|
|
1
|
+
const taskData = {
|
|
2
|
+
datasets: [
|
|
3
|
+
{
|
|
4
|
+
description: "A dataset containing audio conversations with question–answer pairs.",
|
|
5
|
+
id: "nvidia/AF-Think",
|
|
6
|
+
},
|
|
7
|
+
{
|
|
8
|
+
description: "A more advanced and comprehensive dataset that contains characteristics of the audio as well",
|
|
9
|
+
id: "tsinghua-ee/QualiSpeech",
|
|
10
|
+
},
|
|
11
|
+
],
|
|
12
|
+
demo: {
|
|
13
|
+
inputs: [
|
|
14
|
+
{
|
|
15
|
+
filename: "audio.wav",
|
|
16
|
+
type: "audio",
|
|
17
|
+
},
|
|
18
|
+
{
|
|
19
|
+
label: "Text Prompt",
|
|
20
|
+
content: "What is the gender of the speaker?",
|
|
21
|
+
type: "text",
|
|
22
|
+
},
|
|
23
|
+
],
|
|
24
|
+
outputs: [
|
|
25
|
+
{
|
|
26
|
+
label: "Generated Text",
|
|
27
|
+
content: "The gender of the speaker is female.",
|
|
28
|
+
type: "text",
|
|
29
|
+
},
|
|
30
|
+
],
|
|
31
|
+
},
|
|
32
|
+
metrics: [],
|
|
33
|
+
models: [
|
|
34
|
+
{
|
|
35
|
+
description: "A lightweight model that has capabilities of taking both audio and text as inputs and generating responses.",
|
|
36
|
+
id: "fixie-ai/ultravox-v0_5-llama-3_2-1b",
|
|
37
|
+
},
|
|
38
|
+
{
|
|
39
|
+
description: "A multimodal model that supports voice chat and audio analysis.",
|
|
40
|
+
id: "Qwen/Qwen2-Audio-7B-Instruct",
|
|
41
|
+
},
|
|
42
|
+
{
|
|
43
|
+
description: "A model for audio understanding, speech translation, and transcription.",
|
|
44
|
+
id: "mistralai/Voxtral-Small-24B-2507",
|
|
45
|
+
},
|
|
46
|
+
{
|
|
47
|
+
description: "A new model capable of audio question answering and reasoning.",
|
|
48
|
+
id: "nvidia/audio-flamingo-3",
|
|
49
|
+
},
|
|
50
|
+
],
|
|
51
|
+
spaces: [
|
|
52
|
+
{
|
|
53
|
+
description: "A space that takes input as both audio and text and generates answers.",
|
|
54
|
+
id: "iamomtiwari/ATTT",
|
|
55
|
+
},
|
|
56
|
+
{
|
|
57
|
+
description: "A web application that demonstrates chatting with the Qwen2Audio Model.",
|
|
58
|
+
id: "freddyaboulton/talk-to-qwen-webrtc",
|
|
59
|
+
},
|
|
60
|
+
],
|
|
61
|
+
summary: "Audio-text-to-text models take both an audio clip and a text prompt as input, and generate natural language text as output. These models can answer questions about spoken content, summarize meetings, analyze music, or interpret speech beyond simple transcription. They are useful for applications that combine speech understanding with reasoning or conversation.",
|
|
62
|
+
widgetModels: [],
|
|
63
|
+
youtubeId: "",
|
|
64
|
+
};
|
|
65
|
+
export default taskData;
|
package/package.json
CHANGED
package/src/local-apps.ts
CHANGED
|
@@ -90,6 +90,10 @@ function isLlamaCppGgufModel(model: ModelData) {
|
|
|
90
90
|
return !!model.gguf?.context_length;
|
|
91
91
|
}
|
|
92
92
|
|
|
93
|
+
function isAmdRyzenModel(model: ModelData) {
|
|
94
|
+
return model.tags.includes("ryzenai-hybrid") || model.tags.includes("ryzenai-npu");
|
|
95
|
+
}
|
|
96
|
+
|
|
93
97
|
function isMlxModel(model: ModelData) {
|
|
94
98
|
return model.tags.includes("mlx");
|
|
95
99
|
}
|
|
@@ -317,21 +321,38 @@ const snippetDockerModelRunner = (model: ModelData, filepath?: string): string =
|
|
|
317
321
|
|
|
318
322
|
const snippetLemonade = (model: ModelData, filepath?: string): LocalAppSnippet[] => {
|
|
319
323
|
const tagName = getQuantTag(filepath);
|
|
320
|
-
const modelName = model.id.split("/")[1];
|
|
324
|
+
const modelName = model.id.includes("/") ? model.id.split("/")[1] : model.id;
|
|
325
|
+
|
|
326
|
+
// Get recipe according to model type
|
|
327
|
+
let simplifiedModelName: string;
|
|
328
|
+
let recipe: string;
|
|
329
|
+
let checkpoint: string;
|
|
330
|
+
let requirements: string;
|
|
331
|
+
if (model.tags.some((tag) => ["ryzenai-npu", "ryzenai-hybrid"].includes(tag))) {
|
|
332
|
+
recipe = model.tags.includes("ryzenai-npu") ? "oga-npu" : "oga-hybrid";
|
|
333
|
+
checkpoint = model.id;
|
|
334
|
+
requirements = " (requires RyzenAI 300 series)";
|
|
335
|
+
simplifiedModelName = modelName.split("-awq-")[0];
|
|
336
|
+
simplifiedModelName += recipe === "oga-npu" ? "-NPU" : "-Hybrid";
|
|
337
|
+
} else {
|
|
338
|
+
recipe = "llamacpp";
|
|
339
|
+
checkpoint = `${model.id}${tagName}`;
|
|
340
|
+
requirements = "";
|
|
341
|
+
simplifiedModelName = modelName;
|
|
342
|
+
}
|
|
343
|
+
|
|
321
344
|
return [
|
|
322
345
|
{
|
|
323
346
|
title: "Pull the model",
|
|
324
347
|
setup: "# Download Lemonade from https://lemonade-server.ai/",
|
|
325
348
|
content: [
|
|
326
|
-
`lemonade-server pull user.${
|
|
327
|
-
--checkpoint ${model.id}${tagName} \\
|
|
328
|
-
--recipe llamacpp`,
|
|
349
|
+
`lemonade-server pull user.${simplifiedModelName} --checkpoint ${checkpoint} --recipe ${recipe}`,
|
|
329
350
|
"# Note: If you installed from source, use the lemonade-server-dev command instead.",
|
|
330
351
|
].join("\n"),
|
|
331
352
|
},
|
|
332
353
|
{
|
|
333
|
-
title:
|
|
334
|
-
content: `lemonade-server run user.${
|
|
354
|
+
title: `Run and chat with the model${requirements}`,
|
|
355
|
+
content: `lemonade-server run user.${simplifiedModelName}`,
|
|
335
356
|
},
|
|
336
357
|
{
|
|
337
358
|
title: "List all available models",
|
|
@@ -521,7 +542,7 @@ export const LOCAL_APPS = {
|
|
|
521
542
|
prettyLabel: "Lemonade",
|
|
522
543
|
docsUrl: "https://lemonade-server.ai",
|
|
523
544
|
mainTask: "text-generation",
|
|
524
|
-
displayOnModelPage: isLlamaCppGgufModel,
|
|
545
|
+
displayOnModelPage: (model) => isLlamaCppGgufModel(model) || isAmdRyzenModel(model),
|
|
525
546
|
snippet: snippetLemonade,
|
|
526
547
|
},
|
|
527
548
|
} satisfies Record<string, LocalApp>;
|
package/src/model-data.ts
CHANGED
|
@@ -44,6 +44,10 @@ export interface ModelData {
|
|
|
44
44
|
quant_method?: string;
|
|
45
45
|
};
|
|
46
46
|
tokenizer_config?: TokenizerConfig;
|
|
47
|
+
processor_config?: {
|
|
48
|
+
chat_template?: string;
|
|
49
|
+
};
|
|
50
|
+
chat_template_jinja?: string;
|
|
47
51
|
adapter_transformers?: {
|
|
48
52
|
model_name?: string;
|
|
49
53
|
model_class?: string;
|
|
@@ -1476,6 +1476,11 @@ export const terratorch = (model: ModelData): string[] => [
|
|
|
1476
1476
|
model = BACKBONE_REGISTRY.build("${model.id}")`,
|
|
1477
1477
|
];
|
|
1478
1478
|
|
|
1479
|
+
const hasChatTemplate = (model: ModelData): boolean =>
|
|
1480
|
+
model.config?.tokenizer_config?.chat_template !== undefined ||
|
|
1481
|
+
model.config?.processor_config?.chat_template !== undefined ||
|
|
1482
|
+
model.config?.chat_template_jinja !== undefined;
|
|
1483
|
+
|
|
1479
1484
|
export const transformers = (model: ModelData): string[] => {
|
|
1480
1485
|
const info = model.transformersInfo;
|
|
1481
1486
|
if (!info) {
|
|
@@ -1498,7 +1503,7 @@ export const transformers = (model: ModelData): string[] => {
|
|
|
1498
1503
|
`${processorVarName} = ${info.processor}.from_pretrained("${model.id}"` + remote_code_snippet + ")",
|
|
1499
1504
|
`model = ${info.auto_model}.from_pretrained("${model.id}"` + remote_code_snippet + ")"
|
|
1500
1505
|
);
|
|
1501
|
-
if (model.tags.includes("conversational")) {
|
|
1506
|
+
if (model.tags.includes("conversational") && hasChatTemplate(model)) {
|
|
1502
1507
|
if (model.tags.includes("image-text-to-text")) {
|
|
1503
1508
|
autoSnippet.push(
|
|
1504
1509
|
"messages = [",
|
|
@@ -1905,6 +1910,87 @@ export const model2vec = (model: ModelData): string[] => [
|
|
|
1905
1910
|
model = StaticModel.from_pretrained("${model.id}")`,
|
|
1906
1911
|
];
|
|
1907
1912
|
|
|
1913
|
+
export const pruna = (model: ModelData): string[] => {
|
|
1914
|
+
let snippets: string[];
|
|
1915
|
+
|
|
1916
|
+
if (model.tags.includes("diffusers")) {
|
|
1917
|
+
snippets = pruna_diffusers(model);
|
|
1918
|
+
} else if (model.tags.includes("transformers")) {
|
|
1919
|
+
snippets = pruna_transformers(model);
|
|
1920
|
+
} else {
|
|
1921
|
+
snippets = pruna_default(model);
|
|
1922
|
+
}
|
|
1923
|
+
|
|
1924
|
+
const ensurePrunaModelImport = (snippet: string): string => {
|
|
1925
|
+
if (!/^from pruna import PrunaModel/m.test(snippet)) {
|
|
1926
|
+
return `from pruna import PrunaModel\n${snippet}`;
|
|
1927
|
+
}
|
|
1928
|
+
return snippet;
|
|
1929
|
+
};
|
|
1930
|
+
snippets = snippets.map(ensurePrunaModelImport);
|
|
1931
|
+
|
|
1932
|
+
if (model.tags.includes("pruna_pro-ai")) {
|
|
1933
|
+
return snippets.map((snippet) =>
|
|
1934
|
+
snippet.replace(/\bpruna\b/g, "pruna_pro").replace(/\bPrunaModel\b/g, "PrunaProModel")
|
|
1935
|
+
);
|
|
1936
|
+
}
|
|
1937
|
+
|
|
1938
|
+
return snippets;
|
|
1939
|
+
};
|
|
1940
|
+
|
|
1941
|
+
const pruna_diffusers = (model: ModelData): string[] => {
|
|
1942
|
+
const diffusersSnippets = diffusers(model);
|
|
1943
|
+
|
|
1944
|
+
return diffusersSnippets.map((snippet) =>
|
|
1945
|
+
snippet
|
|
1946
|
+
// Replace pipeline classes with PrunaModel
|
|
1947
|
+
.replace(/\b\w*Pipeline\w*\b/g, "PrunaModel")
|
|
1948
|
+
// Clean up diffusers imports containing PrunaModel
|
|
1949
|
+
.replace(/from diffusers import ([^,\n]*PrunaModel[^,\n]*)/g, "")
|
|
1950
|
+
.replace(/from diffusers import ([^,\n]+),?\s*([^,\n]*PrunaModel[^,\n]*)/g, "from diffusers import $1")
|
|
1951
|
+
.replace(/from diffusers import\s*(\n|$)/g, "")
|
|
1952
|
+
// Fix PrunaModel imports
|
|
1953
|
+
.replace(/from diffusers import PrunaModel/g, "from pruna import PrunaModel")
|
|
1954
|
+
.replace(/from diffusers import ([^,\n]+), PrunaModel/g, "from diffusers import $1")
|
|
1955
|
+
.replace(/from diffusers import PrunaModel, ([^,\n]+)/g, "from diffusers import $1")
|
|
1956
|
+
// Clean up whitespace
|
|
1957
|
+
.replace(/\n\n+/g, "\n")
|
|
1958
|
+
.trim()
|
|
1959
|
+
);
|
|
1960
|
+
};
|
|
1961
|
+
|
|
1962
|
+
const pruna_transformers = (model: ModelData): string[] => {
|
|
1963
|
+
const info = model.transformersInfo;
|
|
1964
|
+
const transformersSnippets = transformers(model);
|
|
1965
|
+
|
|
1966
|
+
// Replace pipeline with PrunaModel
|
|
1967
|
+
let processedSnippets = transformersSnippets.map((snippet) =>
|
|
1968
|
+
snippet
|
|
1969
|
+
.replace(/from transformers import pipeline/g, "from pruna import PrunaModel")
|
|
1970
|
+
.replace(/pipeline\([^)]*\)/g, `PrunaModel.from_pretrained("${model.id}")`)
|
|
1971
|
+
);
|
|
1972
|
+
|
|
1973
|
+
// Additional cleanup if auto_model info is available
|
|
1974
|
+
if (info?.auto_model) {
|
|
1975
|
+
processedSnippets = processedSnippets.map((snippet) =>
|
|
1976
|
+
snippet
|
|
1977
|
+
.replace(new RegExp(`from transformers import ${info.auto_model}\n?`, "g"), "")
|
|
1978
|
+
.replace(new RegExp(`${info.auto_model}.from_pretrained`, "g"), "PrunaModel.from_pretrained")
|
|
1979
|
+
.replace(new RegExp(`^.*from.*import.*(, *${info.auto_model})+.*$`, "gm"), (line) =>
|
|
1980
|
+
line.replace(new RegExp(`, *${info.auto_model}`, "g"), "")
|
|
1981
|
+
)
|
|
1982
|
+
);
|
|
1983
|
+
}
|
|
1984
|
+
|
|
1985
|
+
return processedSnippets;
|
|
1986
|
+
};
|
|
1987
|
+
|
|
1988
|
+
const pruna_default = (model: ModelData): string[] => [
|
|
1989
|
+
`from pruna import PrunaModel
|
|
1990
|
+
model = PrunaModel.from_pretrained("${model.id}")
|
|
1991
|
+
`,
|
|
1992
|
+
];
|
|
1993
|
+
|
|
1908
1994
|
export const nemo = (model: ModelData): string[] => {
|
|
1909
1995
|
let command: string[] | undefined = undefined;
|
|
1910
1996
|
// Resolve the tag to a nemo domain/sub-domain
|
package/src/model-libraries.ts
CHANGED
|
@@ -801,6 +801,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
|
|
|
801
801
|
filter: false,
|
|
802
802
|
countDownloads: `path_extension:"pth"`,
|
|
803
803
|
},
|
|
804
|
+
"pruna-ai": {
|
|
805
|
+
prettyLabel: "Pruna AI",
|
|
806
|
+
repoName: "Pruna AI",
|
|
807
|
+
repoUrl: "https://github.com/PrunaAI/pruna",
|
|
808
|
+
snippets: snippets.pruna,
|
|
809
|
+
docsUrl: "https://docs.pruna.ai",
|
|
810
|
+
},
|
|
804
811
|
pxia: {
|
|
805
812
|
prettyLabel: "pxia",
|
|
806
813
|
repoName: "pxia",
|
|
@@ -1072,7 +1079,7 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
|
|
|
1072
1079
|
repoUrl: "https://github.com/IBM/terratorch",
|
|
1073
1080
|
docsUrl: "https://ibm.github.io/terratorch/",
|
|
1074
1081
|
filter: false,
|
|
1075
|
-
countDownloads: `path_extension:"pt"`,
|
|
1082
|
+
countDownloads: `path_extension:"pt" OR path_extension:"ckpt"`,
|
|
1076
1083
|
snippets: snippets.terratorch,
|
|
1077
1084
|
},
|
|
1078
1085
|
"tic-clip": {
|
|
@@ -0,0 +1,139 @@
|
|
|
1
|
+
## Use Cases
|
|
2
|
+
|
|
3
|
+
> This task takes `audio` and a `text prompt` and returns `text` (answers, summaries, structured notes, etc.).
|
|
4
|
+
|
|
5
|
+
### Audio question answering
|
|
6
|
+
|
|
7
|
+
Ask targeted questions about lectures, podcasts, or calls and get context-aware answers.
|
|
8
|
+
**Example:** Audio: physics lecture → Prompt: “What did the teacher say about gravity and how is it measured?”
|
|
9
|
+
|
|
10
|
+
### Meeting notes & action items
|
|
11
|
+
|
|
12
|
+
Turn multi-speaker meetings into concise minutes with decisions, owners, and deadlines.
|
|
13
|
+
**Example:** Audio: weekly stand-up → Prompt: “Summarize key decisions and list action items with assignees.”
|
|
14
|
+
|
|
15
|
+
### Speech understanding & intent
|
|
16
|
+
|
|
17
|
+
Go beyond transcription to extract intent, sentiment, uncertainty, or emotion from spoken language.
|
|
18
|
+
**Example:** “I’m not sure I can finish this on time.” → Prompt: “Describe speaker intent and confidence.”
|
|
19
|
+
|
|
20
|
+
### Music & sound analysis (textual)
|
|
21
|
+
|
|
22
|
+
Describe instrumentation, genre, tempo, or sections, and suggest edits or techniques (text output only).
|
|
23
|
+
**Example:** Song demo → Prompt: “Identify key and tempo, then suggest jazz reharmonization ideas for the chorus.”
|
|
24
|
+
|
|
25
|
+
## Inference
|
|
26
|
+
|
|
27
|
+
You can use the 'transformers' library, and your audio file to any of the `audio-text-to-text` model, with instructions and get text responses. Following code examples show how to do so.
|
|
28
|
+
|
|
29
|
+
### Speech Transcription and Analysis
|
|
30
|
+
|
|
31
|
+
These models don’t just turn speech into text—they also capture tone, emotion, and speaker traits. This makes them useful for tasks like sentiment analysis or identifying speaker profiles.
|
|
32
|
+
|
|
33
|
+
You can try audio transcription with [Voxtral Mini](https://huggingface.co/mistralai/Voxtral-Mini-3B-2507) using the following code.
|
|
34
|
+
|
|
35
|
+
```python
|
|
36
|
+
from transformers import VoxtralForConditionalGeneration, AutoProcessor
|
|
37
|
+
import torch
|
|
38
|
+
|
|
39
|
+
device = "cuda"
|
|
40
|
+
repo_id = "mistralai/Voxtral-Mini-3B-2507"
|
|
41
|
+
|
|
42
|
+
processor = AutoProcessor.from_pretrained(repo_id)
|
|
43
|
+
model = VoxtralForConditionalGeneration.from_pretrained(repo_id, dtype=torch.bfloat16, device_map=device)
|
|
44
|
+
|
|
45
|
+
inputs = processor.apply_transcription_request(language="en", audio="https://huggingface.co/datasets/hf-internal-testing/dummy-audio-samples/resolve/main/obama.mp3", model_id=repo_id)
|
|
46
|
+
inputs = inputs.to(device, dtype=torch.bfloat16)
|
|
47
|
+
|
|
48
|
+
outputs = model.generate(**inputs, max_new_tokens=500)
|
|
49
|
+
decoded_outputs = processor.batch_decode(outputs[:, inputs.input_ids.shape[1]:], skip_special_tokens=True)
|
|
50
|
+
|
|
51
|
+
print("\nGenerated responses:")
|
|
52
|
+
print("=" * 80)
|
|
53
|
+
for decoded_output in decoded_outputs:
|
|
54
|
+
print(decoded_output)
|
|
55
|
+
print("=" * 80)
|
|
56
|
+
```
|
|
57
|
+
|
|
58
|
+
### Audio Question Answering
|
|
59
|
+
|
|
60
|
+
These models can understand audio directly and answer questions about it. For example, summarizing a podcast clip or explaining parts of a recorded conversation.
|
|
61
|
+
|
|
62
|
+
You can experiment with [Qwen2-Audio-Instruct-Demo](https://huggingface.co/Qwen/Qwen2-Audio-Instruct-Demo) for conversations with both text and audio inputs, letting you ask follow-up questions about different sounds or speech clips.
|
|
63
|
+
|
|
64
|
+
```python
|
|
65
|
+
from io import BytesIO
|
|
66
|
+
from urllib.request import urlopen
|
|
67
|
+
import librosa
|
|
68
|
+
from transformers import Qwen2AudioForConditionalGeneration, AutoProcessor
|
|
69
|
+
|
|
70
|
+
processor = AutoProcessor.from_pretrained("Qwen/Qwen2-Audio-7B-Instruct")
|
|
71
|
+
model = Qwen2AudioForConditionalGeneration.from_pretrained("Qwen/Qwen2-Audio-7B-Instruct", device_map="auto")
|
|
72
|
+
|
|
73
|
+
conversation = [
|
|
74
|
+
{'role': 'system', 'content': 'You are a helpful assistant.'},
|
|
75
|
+
{"role": "user", "content": [
|
|
76
|
+
{"type": "audio", "audio_url": "https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen2-Audio/audio/glass-breaking-151256.mp3"},
|
|
77
|
+
{"type": "text", "text": "What's that sound?"},
|
|
78
|
+
]},
|
|
79
|
+
{"role": "assistant", "content": "It is the sound of glass shattering."},
|
|
80
|
+
{"role": "user", "content": [
|
|
81
|
+
{"type": "text", "text": "What can you do when you hear that?"},
|
|
82
|
+
]},
|
|
83
|
+
{"role": "assistant", "content": "Stay alert and cautious, and check if anyone is hurt or if there is any damage to property."},
|
|
84
|
+
{"role": "user", "content": [
|
|
85
|
+
{"type": "audio", "audio_url": "https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen2-Audio/audio/1272-128104-0000.flac"},
|
|
86
|
+
{"type": "text", "text": "What does the person say?"},
|
|
87
|
+
]},
|
|
88
|
+
]
|
|
89
|
+
text = processor.apply_chat_template(conversation, add_generation_prompt=True, tokenize=False)
|
|
90
|
+
audios = []
|
|
91
|
+
for message in conversation:
|
|
92
|
+
if isinstance(message["content"], list):
|
|
93
|
+
for ele in message["content"]:
|
|
94
|
+
if ele["type"] == "audio":
|
|
95
|
+
audios.append(
|
|
96
|
+
librosa.load(
|
|
97
|
+
BytesIO(urlopen(ele['audio_url']).read()),
|
|
98
|
+
sr=processor.feature_extractor.sampling_rate)[0]
|
|
99
|
+
)
|
|
100
|
+
|
|
101
|
+
inputs = processor(text=text, audios=audios, return_tensors="pt", padding=True)
|
|
102
|
+
inputs.input_ids = inputs.input_ids.to("cuda")
|
|
103
|
+
|
|
104
|
+
generate_ids = model.generate(**inputs, max_length=256)
|
|
105
|
+
generate_ids = generate_ids[:, inputs.input_ids.size(1):]
|
|
106
|
+
|
|
107
|
+
response = processor.batch_decode(generate_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
|
|
108
|
+
```
|
|
109
|
+
|
|
110
|
+
## Useful Resources
|
|
111
|
+
|
|
112
|
+
If you want to learn more about this concept, here are some useful links:
|
|
113
|
+
|
|
114
|
+
### Papers
|
|
115
|
+
|
|
116
|
+
- [SpeechGPT](https://huggingface.co/papers/2507.13264) — multimodal dialogue with speech and text.
|
|
117
|
+
- [Voxtral](https://huggingface.co/papers/2507.13264) — a state-of-the-art audio-text model.
|
|
118
|
+
- [Qwen2-audio-instruct](https://huggingface.co/papers/2407.10759) — large-scale audio-language modeling for instruction following.
|
|
119
|
+
- [AudioPaLM](https://huggingface.co/papers/2306.12925) — scaling audio-language models with PaLM.
|
|
120
|
+
|
|
121
|
+
### Models, Codes & Demos
|
|
122
|
+
|
|
123
|
+
- [Qwen2-audio-instruct](https://github.com/QwenLM/Qwen2-Audio) — open-source implementation with demos.
|
|
124
|
+
- [SpeechGPT](https://github.com/0nutation/SpeechGPT) — An end-to-end framework for audio conversational models built on top of large language models.
|
|
125
|
+
- [AudioPaLM](https://google-research.github.io/seanet/audiopalm/examples/) — resources and code for AudioPaLM.
|
|
126
|
+
- [Audio Flamingo](https://huggingface.co/nvidia/audio-flamingo-3) — unifies speech, sound, and music understanding with long-context reasoning.
|
|
127
|
+
- [Ultravox](https://github.com/fixie-ai/ultravox) — a fast multimodal large language model designed for real-time voice interactions.
|
|
128
|
+
- [Ichigo](https://github.com/menloresearch/ichigo) — an audio-text-to-text model for audio-related tasks.
|
|
129
|
+
|
|
130
|
+
### Datasets
|
|
131
|
+
|
|
132
|
+
- [nvidia/AF-Think](https://huggingface.co/datasets/nvidia/AF-Think)
|
|
133
|
+
- [nvidia/AudioSkills](https://huggingface.co/datasets/nvidia/AudioSkills)
|
|
134
|
+
|
|
135
|
+
### Tools & Extras
|
|
136
|
+
|
|
137
|
+
- [Fast-RTC](https://huggingface.co/fastrtc) — turn any Python function into a real-time audio/video stream.
|
|
138
|
+
- [PhiCookBook](https://github.com/microsoft/PhiCookBook) — Microsoft’s open-source guide to small language models.
|
|
139
|
+
- [Qwen2-audio-instruct](https://qwenlm.github.io/blog/qwen2-audio/) — Blogpost explaining usage and demos of Qwen2-audio-instruct.
|
|
@@ -0,0 +1,70 @@
|
|
|
1
|
+
import type { TaskDataCustom } from "../index.js";
|
|
2
|
+
|
|
3
|
+
const taskData: TaskDataCustom = {
|
|
4
|
+
datasets: [
|
|
5
|
+
{
|
|
6
|
+
description: "A dataset containing audio conversations with question–answer pairs.",
|
|
7
|
+
id: "nvidia/AF-Think",
|
|
8
|
+
},
|
|
9
|
+
{
|
|
10
|
+
description: "A more advanced and comprehensive dataset that contains characteristics of the audio as well",
|
|
11
|
+
id: "tsinghua-ee/QualiSpeech",
|
|
12
|
+
},
|
|
13
|
+
],
|
|
14
|
+
demo: {
|
|
15
|
+
inputs: [
|
|
16
|
+
{
|
|
17
|
+
filename: "audio.wav",
|
|
18
|
+
type: "audio",
|
|
19
|
+
},
|
|
20
|
+
{
|
|
21
|
+
label: "Text Prompt",
|
|
22
|
+
content: "What is the gender of the speaker?",
|
|
23
|
+
type: "text",
|
|
24
|
+
},
|
|
25
|
+
],
|
|
26
|
+
outputs: [
|
|
27
|
+
{
|
|
28
|
+
label: "Generated Text",
|
|
29
|
+
content: "The gender of the speaker is female.",
|
|
30
|
+
type: "text",
|
|
31
|
+
},
|
|
32
|
+
],
|
|
33
|
+
},
|
|
34
|
+
metrics: [],
|
|
35
|
+
models: [
|
|
36
|
+
{
|
|
37
|
+
description:
|
|
38
|
+
"A lightweight model that has capabilities of taking both audio and text as inputs and generating responses.",
|
|
39
|
+
id: "fixie-ai/ultravox-v0_5-llama-3_2-1b",
|
|
40
|
+
},
|
|
41
|
+
{
|
|
42
|
+
description: "A multimodal model that supports voice chat and audio analysis.",
|
|
43
|
+
id: "Qwen/Qwen2-Audio-7B-Instruct",
|
|
44
|
+
},
|
|
45
|
+
{
|
|
46
|
+
description: "A model for audio understanding, speech translation, and transcription.",
|
|
47
|
+
id: "mistralai/Voxtral-Small-24B-2507",
|
|
48
|
+
},
|
|
49
|
+
{
|
|
50
|
+
description: "A new model capable of audio question answering and reasoning.",
|
|
51
|
+
id: "nvidia/audio-flamingo-3",
|
|
52
|
+
},
|
|
53
|
+
],
|
|
54
|
+
spaces: [
|
|
55
|
+
{
|
|
56
|
+
description: "A space that takes input as both audio and text and generates answers.",
|
|
57
|
+
id: "iamomtiwari/ATTT",
|
|
58
|
+
},
|
|
59
|
+
{
|
|
60
|
+
description: "A web application that demonstrates chatting with the Qwen2Audio Model.",
|
|
61
|
+
id: "freddyaboulton/talk-to-qwen-webrtc",
|
|
62
|
+
},
|
|
63
|
+
],
|
|
64
|
+
summary:
|
|
65
|
+
"Audio-text-to-text models take both an audio clip and a text prompt as input, and generate natural language text as output. These models can answer questions about spoken content, summarize meetings, analyze music, or interpret speech beyond simple transcription. They are useful for applications that combine speech understanding with reasoning or conversation.",
|
|
66
|
+
widgetModels: [],
|
|
67
|
+
youtubeId: "",
|
|
68
|
+
};
|
|
69
|
+
|
|
70
|
+
export default taskData;
|