@huggingface/tasks 0.10.22 → 0.11.0

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/index.cjs CHANGED
@@ -24,7 +24,6 @@ __export(src_exports, {
24
24
  ALL_MODEL_LIBRARY_KEYS: () => ALL_MODEL_LIBRARY_KEYS,
25
25
  DATASET_LIBRARIES_UI_ELEMENTS: () => DATASET_LIBRARIES_UI_ELEMENTS,
26
26
  DEFAULT_MEMORY_OPTIONS: () => DEFAULT_MEMORY_OPTIONS,
27
- InferenceDisplayability: () => InferenceDisplayability,
28
27
  LIBRARY_TASK_MAPPING: () => LIBRARY_TASK_MAPPING,
29
28
  LOCAL_APPS: () => LOCAL_APPS,
30
29
  MAPPING_DEFAULT_WIDGET: () => MAPPING_DEFAULT_WIDGET,
@@ -2141,7 +2140,7 @@ var taskData11 = {
2141
2140
  demo: {
2142
2141
  inputs: [
2143
2142
  {
2144
- filename: "mask-generation-input.png",
2143
+ filename: "image-text-to-text-input.png",
2145
2144
  type: "img"
2146
2145
  },
2147
2146
  {
@@ -4330,6 +4329,15 @@ var mars5_tts = (model) => [
4330
4329
  from inference import Mars5TTS
4331
4330
  mars5 = Mars5TTS.from_pretrained("${model.id}")`
4332
4331
  ];
4332
+ var mesh_anything = () => [
4333
+ `# Install from https://github.com/buaacyw/MeshAnything.git
4334
+
4335
+ from MeshAnything.models.meshanything import MeshAnything
4336
+
4337
+ # refer to https://github.com/buaacyw/MeshAnything/blob/main/main.py#L91 on how to define args
4338
+ # and https://github.com/buaacyw/MeshAnything/blob/main/app.py regarding usage
4339
+ model = MeshAnything(args)`
4340
+ ];
4333
4341
  var open_clip = (model) => [
4334
4342
  `import open_clip
4335
4343
 
@@ -4858,6 +4866,12 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
4858
4866
  filter: false,
4859
4867
  countDownloads: `path:"params.index.json"`
4860
4868
  },
4869
+ champ: {
4870
+ prettyLabel: "Champ",
4871
+ repoName: "Champ",
4872
+ repoUrl: "https://github.com/fudan-generative-vision/champ",
4873
+ countDownloads: `path:"champ/motion_module.pth"`
4874
+ },
4861
4875
  chat_tts: {
4862
4876
  prettyLabel: "ChatTTS",
4863
4877
  repoName: "ChatTTS",
@@ -4866,6 +4880,13 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
4866
4880
  filter: false,
4867
4881
  countDownloads: `path:"asset/GPT.pt"`
4868
4882
  },
4883
+ colpali: {
4884
+ prettyLabel: "ColPali",
4885
+ repoName: "ColPali",
4886
+ repoUrl: "https://github.com/ManuelFay/colpali",
4887
+ filter: false,
4888
+ countDownloads: `path:"adapter_config.json"`
4889
+ },
4869
4890
  diffusers: {
4870
4891
  prettyLabel: "Diffusers",
4871
4892
  repoName: "\u{1F917}/diffusers",
@@ -4965,6 +4986,12 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
4965
4986
  filter: false,
4966
4987
  countDownloads: `path:"ckpt/tensor00000_000" OR path:"ckpt-0/tensor00000_000"`
4967
4988
  },
4989
+ hallo: {
4990
+ prettyLabel: "Hallo",
4991
+ repoName: "Hallo",
4992
+ repoUrl: "https://github.com/fudan-generative-vision/hallo",
4993
+ countDownloads: `path:"hallo/net.pth"`
4994
+ },
4968
4995
  "hunyuan-dit": {
4969
4996
  prettyLabel: "HunyuanDiT",
4970
4997
  repoName: "HunyuanDiT",
@@ -5015,6 +5042,14 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
5015
5042
  countDownloads: `path:"mars5_ar.safetensors"`,
5016
5043
  snippets: mars5_tts
5017
5044
  },
5045
+ "mesh-anything": {
5046
+ prettyLabel: "MeshAnything",
5047
+ repoName: "MeshAnything",
5048
+ repoUrl: "https://github.com/buaacyw/MeshAnything",
5049
+ filter: false,
5050
+ countDownloads: `path:"MeshAnything_350m.pth"`,
5051
+ snippets: mesh_anything
5052
+ },
5018
5053
  "ml-agents": {
5019
5054
  prettyLabel: "ml-agents",
5020
5055
  repoName: "ml-agents",
@@ -5040,6 +5075,14 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
5040
5075
  filter: false,
5041
5076
  countDownloads: `path:"model.safetensors"`
5042
5077
  },
5078
+ "mlc-llm": {
5079
+ prettyLabel: "MLC-LLM",
5080
+ repoName: "MLC-LLM",
5081
+ repoUrl: "https://github.com/mlc-ai/mlc-llm",
5082
+ docsUrl: "https://llm.mlc.ai/docs/",
5083
+ filter: false,
5084
+ countDownloads: `path:"mlc-chat-config.json"`
5085
+ },
5043
5086
  nemo: {
5044
5087
  prettyLabel: "NeMo",
5045
5088
  repoName: "NeMo",
@@ -5258,17 +5301,6 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
5258
5301
  var ALL_MODEL_LIBRARY_KEYS = Object.keys(MODEL_LIBRARIES_UI_ELEMENTS);
5259
5302
  var ALL_DISPLAY_MODEL_LIBRARY_KEYS = Object.entries(MODEL_LIBRARIES_UI_ELEMENTS).filter(([_, v]) => v.filter).map(([k]) => k);
5260
5303
 
5261
- // src/model-data.ts
5262
- var InferenceDisplayability = /* @__PURE__ */ ((InferenceDisplayability2) => {
5263
- InferenceDisplayability2["Yes"] = "Yes";
5264
- InferenceDisplayability2["ExplicitOptOut"] = "ExplicitOptOut";
5265
- InferenceDisplayability2["CustomCode"] = "CustomCode";
5266
- InferenceDisplayability2["LibraryNotDetected"] = "LibraryNotDetected";
5267
- InferenceDisplayability2["PipelineNotDetected"] = "PipelineNotDetected";
5268
- InferenceDisplayability2["PipelineLibraryPairNotSupported"] = "PipelineLibraryPairNotSupported";
5269
- return InferenceDisplayability2;
5270
- })(InferenceDisplayability || {});
5271
-
5272
5304
  // src/tokenizer-data.ts
5273
5305
  var SPECIAL_TOKENS_ATTRIBUTES = [
5274
5306
  "bos_token",
@@ -6361,7 +6393,6 @@ var DATASET_LIBRARIES_UI_ELEMENTS = {
6361
6393
  ALL_MODEL_LIBRARY_KEYS,
6362
6394
  DATASET_LIBRARIES_UI_ELEMENTS,
6363
6395
  DEFAULT_MEMORY_OPTIONS,
6364
- InferenceDisplayability,
6365
6396
  LIBRARY_TASK_MAPPING,
6366
6397
  LOCAL_APPS,
6367
6398
  MAPPING_DEFAULT_WIDGET,
package/dist/index.js CHANGED
@@ -2102,7 +2102,7 @@ var taskData11 = {
2102
2102
  demo: {
2103
2103
  inputs: [
2104
2104
  {
2105
- filename: "mask-generation-input.png",
2105
+ filename: "image-text-to-text-input.png",
2106
2106
  type: "img"
2107
2107
  },
2108
2108
  {
@@ -4291,6 +4291,15 @@ var mars5_tts = (model) => [
4291
4291
  from inference import Mars5TTS
4292
4292
  mars5 = Mars5TTS.from_pretrained("${model.id}")`
4293
4293
  ];
4294
+ var mesh_anything = () => [
4295
+ `# Install from https://github.com/buaacyw/MeshAnything.git
4296
+
4297
+ from MeshAnything.models.meshanything import MeshAnything
4298
+
4299
+ # refer to https://github.com/buaacyw/MeshAnything/blob/main/main.py#L91 on how to define args
4300
+ # and https://github.com/buaacyw/MeshAnything/blob/main/app.py regarding usage
4301
+ model = MeshAnything(args)`
4302
+ ];
4294
4303
  var open_clip = (model) => [
4295
4304
  `import open_clip
4296
4305
 
@@ -4819,6 +4828,12 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
4819
4828
  filter: false,
4820
4829
  countDownloads: `path:"params.index.json"`
4821
4830
  },
4831
+ champ: {
4832
+ prettyLabel: "Champ",
4833
+ repoName: "Champ",
4834
+ repoUrl: "https://github.com/fudan-generative-vision/champ",
4835
+ countDownloads: `path:"champ/motion_module.pth"`
4836
+ },
4822
4837
  chat_tts: {
4823
4838
  prettyLabel: "ChatTTS",
4824
4839
  repoName: "ChatTTS",
@@ -4827,6 +4842,13 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
4827
4842
  filter: false,
4828
4843
  countDownloads: `path:"asset/GPT.pt"`
4829
4844
  },
4845
+ colpali: {
4846
+ prettyLabel: "ColPali",
4847
+ repoName: "ColPali",
4848
+ repoUrl: "https://github.com/ManuelFay/colpali",
4849
+ filter: false,
4850
+ countDownloads: `path:"adapter_config.json"`
4851
+ },
4830
4852
  diffusers: {
4831
4853
  prettyLabel: "Diffusers",
4832
4854
  repoName: "\u{1F917}/diffusers",
@@ -4926,6 +4948,12 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
4926
4948
  filter: false,
4927
4949
  countDownloads: `path:"ckpt/tensor00000_000" OR path:"ckpt-0/tensor00000_000"`
4928
4950
  },
4951
+ hallo: {
4952
+ prettyLabel: "Hallo",
4953
+ repoName: "Hallo",
4954
+ repoUrl: "https://github.com/fudan-generative-vision/hallo",
4955
+ countDownloads: `path:"hallo/net.pth"`
4956
+ },
4929
4957
  "hunyuan-dit": {
4930
4958
  prettyLabel: "HunyuanDiT",
4931
4959
  repoName: "HunyuanDiT",
@@ -4976,6 +5004,14 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
4976
5004
  countDownloads: `path:"mars5_ar.safetensors"`,
4977
5005
  snippets: mars5_tts
4978
5006
  },
5007
+ "mesh-anything": {
5008
+ prettyLabel: "MeshAnything",
5009
+ repoName: "MeshAnything",
5010
+ repoUrl: "https://github.com/buaacyw/MeshAnything",
5011
+ filter: false,
5012
+ countDownloads: `path:"MeshAnything_350m.pth"`,
5013
+ snippets: mesh_anything
5014
+ },
4979
5015
  "ml-agents": {
4980
5016
  prettyLabel: "ml-agents",
4981
5017
  repoName: "ml-agents",
@@ -5001,6 +5037,14 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
5001
5037
  filter: false,
5002
5038
  countDownloads: `path:"model.safetensors"`
5003
5039
  },
5040
+ "mlc-llm": {
5041
+ prettyLabel: "MLC-LLM",
5042
+ repoName: "MLC-LLM",
5043
+ repoUrl: "https://github.com/mlc-ai/mlc-llm",
5044
+ docsUrl: "https://llm.mlc.ai/docs/",
5045
+ filter: false,
5046
+ countDownloads: `path:"mlc-chat-config.json"`
5047
+ },
5004
5048
  nemo: {
5005
5049
  prettyLabel: "NeMo",
5006
5050
  repoName: "NeMo",
@@ -5219,17 +5263,6 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
5219
5263
  var ALL_MODEL_LIBRARY_KEYS = Object.keys(MODEL_LIBRARIES_UI_ELEMENTS);
5220
5264
  var ALL_DISPLAY_MODEL_LIBRARY_KEYS = Object.entries(MODEL_LIBRARIES_UI_ELEMENTS).filter(([_, v]) => v.filter).map(([k]) => k);
5221
5265
 
5222
- // src/model-data.ts
5223
- var InferenceDisplayability = /* @__PURE__ */ ((InferenceDisplayability2) => {
5224
- InferenceDisplayability2["Yes"] = "Yes";
5225
- InferenceDisplayability2["ExplicitOptOut"] = "ExplicitOptOut";
5226
- InferenceDisplayability2["CustomCode"] = "CustomCode";
5227
- InferenceDisplayability2["LibraryNotDetected"] = "LibraryNotDetected";
5228
- InferenceDisplayability2["PipelineNotDetected"] = "PipelineNotDetected";
5229
- InferenceDisplayability2["PipelineLibraryPairNotSupported"] = "PipelineLibraryPairNotSupported";
5230
- return InferenceDisplayability2;
5231
- })(InferenceDisplayability || {});
5232
-
5233
5266
  // src/tokenizer-data.ts
5234
5267
  var SPECIAL_TOKENS_ATTRIBUTES = [
5235
5268
  "bos_token",
@@ -6321,7 +6354,6 @@ export {
6321
6354
  ALL_MODEL_LIBRARY_KEYS,
6322
6355
  DATASET_LIBRARIES_UI_ELEMENTS,
6323
6356
  DEFAULT_MEMORY_OPTIONS,
6324
- InferenceDisplayability,
6325
6357
  LIBRARY_TASK_MAPPING,
6326
6358
  LOCAL_APPS,
6327
6359
  MAPPING_DEFAULT_WIDGET,
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=inference-tei-import.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"inference-tei-import.d.ts","sourceRoot":"","sources":["../../scripts/inference-tei-import.ts"],"names":[],"mappings":""}
@@ -8,7 +8,6 @@ export type { LibraryUiElement, ModelLibraryKey } from "./model-libraries";
8
8
  export type { ModelData, TransformersInfo } from "./model-data";
9
9
  export type { AddedToken, SpecialTokensMap, TokenizerConfig } from "./tokenizer-data";
10
10
  export type { WidgetExample, WidgetExampleAttribute, WidgetExampleAssetAndPromptInput, WidgetExampleAssetAndTextInput, WidgetExampleAssetAndZeroShotInput, WidgetExampleAssetInput, WidgetExampleChatInput, WidgetExampleSentenceSimilarityInput, WidgetExampleStructuredDataInput, WidgetExampleTableDataInput, WidgetExampleTextAndContextInput, WidgetExampleTextAndTableInput, WidgetExampleTextInput, WidgetExampleZeroShotTextInput, WidgetExampleOutput, WidgetExampleOutputUrl, WidgetExampleOutputLabels, WidgetExampleOutputAnswerScore, WidgetExampleOutputText, } from "./widget-example";
11
- export { InferenceDisplayability } from "./model-data";
12
11
  export { SPECIAL_TOKENS_ATTRIBUTES } from "./tokenizer-data";
13
12
  import * as snippets from "./snippets";
14
13
  export { snippets };
@@ -1 +1 @@
1
- {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAC1D,OAAO,EAAE,sBAAsB,EAAE,MAAM,yBAAyB,CAAC;AACjE,YAAY,EAAE,QAAQ,EAAE,QAAQ,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AAC9E,cAAc,SAAS,CAAC;AACxB,OAAO,EACN,aAAa,EACb,cAAc,EACd,KAAK,UAAU,EACf,KAAK,YAAY,EACjB,KAAK,YAAY,EACjB,KAAK,QAAQ,EACb,UAAU,EACV,eAAe,EACf,aAAa,EACb,kBAAkB,GAClB,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,8BAA8B,EAAE,sBAAsB,EAAE,2BAA2B,EAAE,MAAM,mBAAmB,CAAC;AACxH,YAAY,EAAE,gBAAgB,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAC3E,YAAY,EAAE,SAAS,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AAChE,YAAY,EAAE,UAAU,EAAE,gBAAgB,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACtF,YAAY,EACX,aAAa,EACb,sBAAsB,EACtB,gCAAgC,EAChC,8BAA8B,EAC9B,kCAAkC,EAClC,uBAAuB,EACvB,sBAAsB,EACtB,oCAAoC,EACpC,gCAAgC,EAChC,2BAA2B,EAC3B,gCAAgC,EAChC,8BAA8B,EAC9B,sBAAsB,EACtB,8BAA8B,EAC9B,mBAAmB,EACnB,sBAAsB,EACtB,yBAAyB,EACzB,8BAA8B,EAC9B,uBAAuB,GACvB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,uBAAuB,EAAE,MAAM,cAAc,CAAC;AACvD,OAAO,EAAE,yBAAyB,EAAE,MAAM,kBAAkB,CAAC;AAE7D,OAAO,KAAK,QAAQ,MAAM,YAAY,CAAC;AACvC,OAAO,EAAE,QAAQ,EAAE,CAAC;AAEpB,OAAO,EAAE,IAAI,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC;AAC1D,YAAY,EAAE,YAAY,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AACxD,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC1C,YAAY,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAE1D,OAAO,EAAE,6BAA6B,EAAE,MAAM,qBAAqB,CAAC;AACpE,YAAY,EAAE,uBAAuB,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC"}
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAC1D,OAAO,EAAE,sBAAsB,EAAE,MAAM,yBAAyB,CAAC;AACjE,YAAY,EAAE,QAAQ,EAAE,QAAQ,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AAC9E,cAAc,SAAS,CAAC;AACxB,OAAO,EACN,aAAa,EACb,cAAc,EACd,KAAK,UAAU,EACf,KAAK,YAAY,EACjB,KAAK,YAAY,EACjB,KAAK,QAAQ,EACb,UAAU,EACV,eAAe,EACf,aAAa,EACb,kBAAkB,GAClB,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,8BAA8B,EAAE,sBAAsB,EAAE,2BAA2B,EAAE,MAAM,mBAAmB,CAAC;AACxH,YAAY,EAAE,gBAAgB,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAC3E,YAAY,EAAE,SAAS,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AAChE,YAAY,EAAE,UAAU,EAAE,gBAAgB,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACtF,YAAY,EACX,aAAa,EACb,sBAAsB,EACtB,gCAAgC,EAChC,8BAA8B,EAC9B,kCAAkC,EAClC,uBAAuB,EACvB,sBAAsB,EACtB,oCAAoC,EACpC,gCAAgC,EAChC,2BAA2B,EAC3B,gCAAgC,EAChC,8BAA8B,EAC9B,sBAAsB,EACtB,8BAA8B,EAC9B,mBAAmB,EACnB,sBAAsB,EACtB,yBAAyB,EACzB,8BAA8B,EAC9B,uBAAuB,GACvB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,yBAAyB,EAAE,MAAM,kBAAkB,CAAC;AAE7D,OAAO,KAAK,QAAQ,MAAM,YAAY,CAAC;AACvC,OAAO,EAAE,QAAQ,EAAE,CAAC;AAEpB,OAAO,EAAE,IAAI,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC;AAC1D,YAAY,EAAE,YAAY,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AACxD,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC1C,YAAY,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAE1D,OAAO,EAAE,6BAA6B,EAAE,MAAM,qBAAqB,CAAC;AACpE,YAAY,EAAE,uBAAuB,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC"}
@@ -1,20 +1,6 @@
1
1
  import type { PipelineType } from "./pipelines";
2
2
  import type { WidgetExample } from "./widget-example";
3
3
  import type { TokenizerConfig } from "./tokenizer-data";
4
- export declare enum InferenceDisplayability {
5
- /**
6
- * Yes
7
- */
8
- Yes = "Yes",
9
- /**
10
- * And then, all the possible reasons why it's no:
11
- */
12
- ExplicitOptOut = "ExplicitOptOut",
13
- CustomCode = "CustomCode",
14
- LibraryNotDetected = "LibraryNotDetected",
15
- PipelineNotDetected = "PipelineNotDetected",
16
- PipelineLibraryPairNotSupported = "PipelineLibraryPairNotSupported"
17
- }
18
4
  /**
19
5
  * Public interface for model metadata
20
6
  */
@@ -25,8 +11,9 @@ export interface ModelData {
25
11
  id: string;
26
12
  /**
27
13
  * Whether or not to enable inference widget for this model
14
+ * TODO(type it)
28
15
  */
29
- inference: InferenceDisplayability;
16
+ inference: string;
30
17
  /**
31
18
  * is this model private?
32
19
  */
@@ -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,aAAa,CAAC;AAChD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAExD,oBAAY,uBAAuB;IAClC;;OAEG;IACH,GAAG,QAAQ;IACX;;OAEG;IACH,cAAc,mBAAmB;IACjC,UAAU,eAAe;IACzB,kBAAkB,uBAAuB;IACzC,mBAAmB,wBAAwB;IAC3C,+BAA+B,oCAAoC;CACnE;AAED;;GAEG;AACH,MAAM,WAAW,SAAS;IACzB;;OAEG;IACH,EAAE,EAAE,MAAM,CAAC;IACX;;OAEG;IACH,SAAS,EAAE,uBAAuB,CAAC;IACnC;;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;SACvB,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;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;KAC/B,CAAC;IACF;;;OAGG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC;CACtB;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,aAAa,CAAC;AAChD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAExD;;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;SACvB,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;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;KAC/B,CAAC;IACF;;;OAGG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC;CACtB;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"}
@@ -17,6 +17,7 @@ export declare const keras: (model: ModelData) => string[];
17
17
  export declare const keras_nlp: (model: ModelData) => string[];
18
18
  export declare const tf_keras: (model: ModelData) => string[];
19
19
  export declare const mars5_tts: (model: ModelData) => string[];
20
+ export declare const mesh_anything: () => string[];
20
21
  export declare const open_clip: (model: ModelData) => string[];
21
22
  export declare const paddlenlp: (model: ModelData) => string[];
22
23
  export declare const pyannote_audio_pipeline: (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,cAAc,CAAC;AAY9C,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAkBF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAkBlD,CAAC;AAMF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AA+BF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAUlD,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAgB/C,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAMlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EASlD,CAAC;AAIF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAO/C,CAAC;AAEF,eAAO,MAAM,OAAO,UAAW,SAAS,KAAG,MAAM,EAMhD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAS9C,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAUlD,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAOjD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAKlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAKlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAsBlD,CAAC;AAEF,eAAO,MAAM,uBAAuB,UAAW,SAAS,KAAG,MAAM,EAehE,CAAC;AAiBF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAKvD,CAAC;AAyBF,eAAO,MAAM,aAAa,UAAW,SAAS,KAAG,MAAM,EAOtD,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAI7C,CAAC;AAsCF,eAAO,MAAM,OAAO,UAAW,SAAS,KAAG,MAAM,EAehD,CAAC;AAEF,eAAO,MAAM,kBAAkB,UAAW,SAAS,KAAG,MAAM,EAmC3D,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,aAAa,UAAW,SAAS,KAAG,MAAM,EAEtD,CAAC;AAEF,eAAO,MAAM,oBAAoB,UAAW,SAAS,KAAG,MAAM,EAI7D,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAU9C,CAAC;AAEF,eAAO,MAAM,WAAW,UAAW,SAAS,KAAG,MAAM,EAIpD,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAK/C,CAAC;AAkBF,eAAO,MAAM,WAAW,UAAW,SAAS,KAAG,MAAM,EAkBpD,CAAC;AAEF,eAAO,MAAM,YAAY,UAAW,SAAS,KAAG,MAAM,EA4CrD,CAAC;AAEF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAcvD,CAAC;AAiBF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAkB7C,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAEF,eAAO,MAAM,gBAAgB,UAAW,SAAS,KAAG,MAAM,EAMzD,CAAC;AAgBF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAEjD,CAAC;AAEF,eAAO,MAAM,MAAM,QAA6B,MAAM,EAMrD,CAAC;AAEF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAInD,CAAC;AAEF,eAAO,MAAM,OAAO,QAAO,MAAM,EAYhC,CAAC;AAEF,eAAO,MAAM,GAAG,UAAW,SAAS,KAAG,MAAM,EAK5C,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAQ7C,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AA6BF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAUnD,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAYnC,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,cAAc,CAAC;AAY9C,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAkBF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAkBlD,CAAC;AAMF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AA+BF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAUlD,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAgB/C,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAMlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EASlD,CAAC;AAIF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAO/C,CAAC;AAEF,eAAO,MAAM,OAAO,UAAW,SAAS,KAAG,MAAM,EAMhD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAS9C,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAUlD,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAOjD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAKlD,CAAC;AAEF,eAAO,MAAM,aAAa,QAAO,MAAM,EAQtC,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAKlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAsBlD,CAAC;AAEF,eAAO,MAAM,uBAAuB,UAAW,SAAS,KAAG,MAAM,EAehE,CAAC;AAiBF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAKvD,CAAC;AAyBF,eAAO,MAAM,aAAa,UAAW,SAAS,KAAG,MAAM,EAOtD,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAI7C,CAAC;AAsCF,eAAO,MAAM,OAAO,UAAW,SAAS,KAAG,MAAM,EAehD,CAAC;AAEF,eAAO,MAAM,kBAAkB,UAAW,SAAS,KAAG,MAAM,EAmC3D,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,aAAa,UAAW,SAAS,KAAG,MAAM,EAEtD,CAAC;AAEF,eAAO,MAAM,oBAAoB,UAAW,SAAS,KAAG,MAAM,EAI7D,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAU9C,CAAC;AAEF,eAAO,MAAM,WAAW,UAAW,SAAS,KAAG,MAAM,EAIpD,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAK/C,CAAC;AAkBF,eAAO,MAAM,WAAW,UAAW,SAAS,KAAG,MAAM,EAkBpD,CAAC;AAEF,eAAO,MAAM,YAAY,UAAW,SAAS,KAAG,MAAM,EA4CrD,CAAC;AAEF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAcvD,CAAC;AAiBF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAkB7C,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAEF,eAAO,MAAM,gBAAgB,UAAW,SAAS,KAAG,MAAM,EAMzD,CAAC;AAgBF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAEjD,CAAC;AAEF,eAAO,MAAM,MAAM,QAA6B,MAAM,EAMrD,CAAC;AAEF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAInD,CAAC;AAEF,eAAO,MAAM,OAAO,QAAO,MAAM,EAYhC,CAAC;AAEF,eAAO,MAAM,GAAG,UAAW,SAAS,KAAG,MAAM,EAK5C,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAQ7C,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AA6BF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAUnD,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAYnC,CAAC"}
@@ -118,6 +118,12 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
118
118
  filter: false;
119
119
  countDownloads: string;
120
120
  };
121
+ champ: {
122
+ prettyLabel: string;
123
+ repoName: string;
124
+ repoUrl: string;
125
+ countDownloads: string;
126
+ };
121
127
  chat_tts: {
122
128
  prettyLabel: string;
123
129
  repoName: string;
@@ -126,6 +132,13 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
126
132
  filter: false;
127
133
  countDownloads: string;
128
134
  };
135
+ colpali: {
136
+ prettyLabel: string;
137
+ repoName: string;
138
+ repoUrl: string;
139
+ filter: false;
140
+ countDownloads: string;
141
+ };
129
142
  diffusers: {
130
143
  prettyLabel: string;
131
144
  repoName: string;
@@ -224,6 +237,12 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
224
237
  filter: false;
225
238
  countDownloads: string;
226
239
  };
240
+ hallo: {
241
+ prettyLabel: string;
242
+ repoName: string;
243
+ repoUrl: string;
244
+ countDownloads: string;
245
+ };
227
246
  "hunyuan-dit": {
228
247
  prettyLabel: string;
229
248
  repoName: string;
@@ -273,6 +292,14 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
273
292
  countDownloads: string;
274
293
  snippets: (model: ModelData) => string[];
275
294
  };
295
+ "mesh-anything": {
296
+ prettyLabel: string;
297
+ repoName: string;
298
+ repoUrl: string;
299
+ filter: false;
300
+ countDownloads: string;
301
+ snippets: () => string[];
302
+ };
276
303
  "ml-agents": {
277
304
  prettyLabel: string;
278
305
  repoName: string;
@@ -298,6 +325,14 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
298
325
  filter: false;
299
326
  countDownloads: string;
300
327
  };
328
+ "mlc-llm": {
329
+ prettyLabel: string;
330
+ repoName: string;
331
+ repoUrl: string;
332
+ docsUrl: string;
333
+ filter: false;
334
+ countDownloads: string;
335
+ };
301
336
  nemo: {
302
337
  prettyLabel: string;
303
338
  repoName: string;
@@ -514,6 +549,6 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
514
549
  };
515
550
  };
516
551
  export type ModelLibraryKey = keyof typeof MODEL_LIBRARIES_UI_ELEMENTS;
517
- export declare const ALL_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "audioseal" | "bertopic" | "big_vision" | "bm25s" | "chat_tts" | "diffusers" | "doctr" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gemma.cpp" | "gliner" | "glyph-byt5" | "grok" | "hunyuan-dit" | "keras" | "tf-keras" | "keras-nlp" | "k2" | "mindspore" | "mars5-tts" | "ml-agents" | "mlx" | "mlx-image" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "recurrentgemma" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "stable-audio-tools" | "diffusion-single-file" | "stable-baselines3" | "stanza" | "tensorflowtts" | "tic-clip" | "timesfm" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "voicecraft" | "whisperkit")[];
518
- export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "audioseal" | "bertopic" | "big_vision" | "bm25s" | "chat_tts" | "diffusers" | "doctr" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gemma.cpp" | "gliner" | "glyph-byt5" | "grok" | "hunyuan-dit" | "keras" | "tf-keras" | "keras-nlp" | "k2" | "mindspore" | "mars5-tts" | "ml-agents" | "mlx" | "mlx-image" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "recurrentgemma" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "stable-audio-tools" | "diffusion-single-file" | "stable-baselines3" | "stanza" | "tensorflowtts" | "tic-clip" | "timesfm" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "voicecraft" | "whisperkit")[];
552
+ export declare const ALL_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "audioseal" | "bertopic" | "big_vision" | "bm25s" | "champ" | "chat_tts" | "colpali" | "diffusers" | "doctr" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gemma.cpp" | "gliner" | "glyph-byt5" | "grok" | "hallo" | "hunyuan-dit" | "keras" | "tf-keras" | "keras-nlp" | "k2" | "mindspore" | "mars5-tts" | "mesh-anything" | "ml-agents" | "mlx" | "mlx-image" | "mlc-llm" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "recurrentgemma" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "stable-audio-tools" | "diffusion-single-file" | "stable-baselines3" | "stanza" | "tensorflowtts" | "tic-clip" | "timesfm" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "voicecraft" | "whisperkit")[];
553
+ export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "audioseal" | "bertopic" | "big_vision" | "bm25s" | "champ" | "chat_tts" | "colpali" | "diffusers" | "doctr" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gemma.cpp" | "gliner" | "glyph-byt5" | "grok" | "hallo" | "hunyuan-dit" | "keras" | "tf-keras" | "keras-nlp" | "k2" | "mindspore" | "mars5-tts" | "mesh-anything" | "ml-agents" | "mlx" | "mlx-image" | "mlc-llm" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "recurrentgemma" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "stable-audio-tools" | "diffusion-single-file" | "stable-baselines3" | "stanza" | "tensorflowtts" | "tic-clip" | "timesfm" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "voicecraft" | "whisperkit")[];
519
554
  //# 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,cAAc,CAAC;AAC9C,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,6BAA6B,CAAC;AAEtE;;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;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA6cI,CAAC;AAE7C,MAAM,MAAM,eAAe,GAAG,MAAM,OAAO,2BAA2B,CAAC;AAEvE,eAAO,MAAM,sBAAsB,8zBAAgE,CAAC;AAEpG,eAAO,MAAM,8BAA8B,8zBAQ1B,CAAC"}
1
+ {"version":3,"file":"model-libraries.d.ts","sourceRoot":"","sources":["../../src/model-libraries.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,6BAA6B,CAAC;AAEtE;;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;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAgfI,CAAC;AAE7C,MAAM,MAAM,eAAe,GAAG,MAAM,OAAO,2BAA2B,CAAC;AAEvE,eAAO,MAAM,sBAAsB,43BAAgE,CAAC;AAEpG,eAAO,MAAM,8BAA8B,43BAQ1B,CAAC"}
@@ -3,21 +3,36 @@
3
3
  *
4
4
  * Using src/scripts/inference-codegen
5
5
  */
6
- export type FeatureExtractionOutput = unknown[];
6
+ export type FeatureExtractionOutput = Array<number[]>;
7
7
  /**
8
- * Inputs for Text Embedding inference
8
+ * Feature Extraction Input.
9
+ *
10
+ * Auto-generated from TEI specs.
11
+ * For more details, check out
12
+ * https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-tei-import.ts.
9
13
  */
10
14
  export interface FeatureExtractionInput {
11
15
  /**
12
- * The text to get the embeddings of
16
+ * The text to embed.
13
17
  */
14
18
  inputs: string;
19
+ normalize?: boolean;
15
20
  /**
16
- * Additional inference parameters
21
+ * The name of the prompt that should be used by for encoding. If not set, no prompt
22
+ * will be applied.
23
+ *
24
+ * Must be a key in the `Sentence Transformers` configuration `prompts` dictionary.
25
+ *
26
+ * For example if ``prompt_name`` is "query" and the ``prompts`` is {"query": "query: ",
27
+ * ...},
28
+ * then the sentence "What is the capital of France?" will be encoded as
29
+ * "query: What is the capital of France?" because the prompt text will be prepended before
30
+ * any text to encode.
17
31
  */
18
- parameters?: {
19
- [key: string]: unknown;
20
- };
32
+ prompt_name?: string;
33
+ truncate?: boolean;
34
+ truncation_direction?: FeatureExtractionInputTruncationDirection;
21
35
  [property: string]: unknown;
22
36
  }
37
+ export type FeatureExtractionInputTruncationDirection = "Left" | "Right";
23
38
  //# sourceMappingURL=inference.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../../src/tasks/feature-extraction/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,MAAM,MAAM,uBAAuB,GAAG,OAAO,EAAE,CAAC;AAEhD;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACtC;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,UAAU,CAAC,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;KAAE,CAAC;IACxC,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}
1
+ {"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../../src/tasks/feature-extraction/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,MAAM,MAAM,uBAAuB,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;AAEtD;;;;;;GAMG;AACH,MAAM,WAAW,sBAAsB;IACtC;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB;;;;;;;;;;;OAWG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,oBAAoB,CAAC,EAAE,yCAAyC,CAAC;IACjE,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED,MAAM,MAAM,yCAAyC,GAAG,MAAM,GAAG,OAAO,CAAC"}
package/package.json CHANGED
@@ -1,7 +1,7 @@
1
1
  {
2
2
  "name": "@huggingface/tasks",
3
3
  "packageManager": "pnpm@8.10.5",
4
- "version": "0.10.22",
4
+ "version": "0.11.0",
5
5
  "description": "List of ML tasks for huggingface.co/tasks",
6
6
  "repository": "https://github.com/huggingface/huggingface.js.git",
7
7
  "publishConfig": {
@@ -44,6 +44,7 @@
44
44
  "build": "tsup src/index.ts --format cjs,esm --clean && tsc --emitDeclarationOnly --declaration",
45
45
  "check": "tsc",
46
46
  "inference-codegen": "tsx scripts/inference-codegen.ts && prettier --write src/tasks/*/inference.ts",
47
- "inference-tgi-import": "tsx scripts/inference-tgi-import.ts && prettier --write src/tasks/text-generation/spec/*.json && prettier --write src/tasks/chat-completion/spec/*.json"
47
+ "inference-tgi-import": "tsx scripts/inference-tgi-import.ts && prettier --write src/tasks/text-generation/spec/*.json && prettier --write src/tasks/chat-completion/spec/*.json",
48
+ "inference-tei-import": "tsx scripts/inference-tei-import.ts && prettier --write src/tasks/feature-extraction/spec/*.json"
48
49
  }
49
50
  }
package/src/index.ts CHANGED
@@ -39,7 +39,6 @@ export type {
39
39
  WidgetExampleOutputAnswerScore,
40
40
  WidgetExampleOutputText,
41
41
  } from "./widget-example";
42
- export { InferenceDisplayability } from "./model-data";
43
42
  export { SPECIAL_TOKENS_ATTRIBUTES } from "./tokenizer-data";
44
43
 
45
44
  import * as snippets from "./snippets";
package/src/model-data.ts CHANGED
@@ -2,21 +2,6 @@ import type { PipelineType } from "./pipelines";
2
2
  import type { WidgetExample } from "./widget-example";
3
3
  import type { TokenizerConfig } from "./tokenizer-data";
4
4
 
5
- export enum InferenceDisplayability {
6
- /**
7
- * Yes
8
- */
9
- Yes = "Yes",
10
- /**
11
- * And then, all the possible reasons why it's no:
12
- */
13
- ExplicitOptOut = "ExplicitOptOut",
14
- CustomCode = "CustomCode",
15
- LibraryNotDetected = "LibraryNotDetected",
16
- PipelineNotDetected = "PipelineNotDetected",
17
- PipelineLibraryPairNotSupported = "PipelineLibraryPairNotSupported",
18
- }
19
-
20
5
  /**
21
6
  * Public interface for model metadata
22
7
  */
@@ -27,8 +12,9 @@ export interface ModelData {
27
12
  id: string;
28
13
  /**
29
14
  * Whether or not to enable inference widget for this model
15
+ * TODO(type it)
30
16
  */
31
- inference: InferenceDisplayability;
17
+ inference: string;
32
18
  /**
33
19
  * is this model private?
34
20
  */
@@ -230,6 +230,16 @@ from inference import Mars5TTS
230
230
  mars5 = Mars5TTS.from_pretrained("${model.id}")`,
231
231
  ];
232
232
 
233
+ export const mesh_anything = (): string[] => [
234
+ `# Install from https://github.com/buaacyw/MeshAnything.git
235
+
236
+ from MeshAnything.models.meshanything import MeshAnything
237
+
238
+ # refer to https://github.com/buaacyw/MeshAnything/blob/main/main.py#L91 on how to define args
239
+ # and https://github.com/buaacyw/MeshAnything/blob/main/app.py regarding usage
240
+ model = MeshAnything(args)`,
241
+ ];
242
+
233
243
  export const open_clip = (model: ModelData): string[] => [
234
244
  `import open_clip
235
245
 
@@ -122,6 +122,12 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
122
122
  filter: false,
123
123
  countDownloads: `path:"params.index.json"`,
124
124
  },
125
+ champ: {
126
+ prettyLabel: "Champ",
127
+ repoName: "Champ",
128
+ repoUrl: "https://github.com/fudan-generative-vision/champ",
129
+ countDownloads: `path:"champ/motion_module.pth"`,
130
+ },
125
131
  chat_tts: {
126
132
  prettyLabel: "ChatTTS",
127
133
  repoName: "ChatTTS",
@@ -130,6 +136,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
130
136
  filter: false,
131
137
  countDownloads: `path:"asset/GPT.pt"`,
132
138
  },
139
+ colpali: {
140
+ prettyLabel: "ColPali",
141
+ repoName: "ColPali",
142
+ repoUrl: "https://github.com/ManuelFay/colpali",
143
+ filter: false,
144
+ countDownloads: `path:"adapter_config.json"`,
145
+ },
133
146
  diffusers: {
134
147
  prettyLabel: "Diffusers",
135
148
  repoName: "🤗/diffusers",
@@ -229,6 +242,12 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
229
242
  filter: false,
230
243
  countDownloads: `path:"ckpt/tensor00000_000" OR path:"ckpt-0/tensor00000_000"`,
231
244
  },
245
+ hallo: {
246
+ prettyLabel: "Hallo",
247
+ repoName: "Hallo",
248
+ repoUrl: "https://github.com/fudan-generative-vision/hallo",
249
+ countDownloads: `path:"hallo/net.pth"`,
250
+ },
232
251
  "hunyuan-dit": {
233
252
  prettyLabel: "HunyuanDiT",
234
253
  repoName: "HunyuanDiT",
@@ -279,6 +298,14 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
279
298
  countDownloads: `path:"mars5_ar.safetensors"`,
280
299
  snippets: snippets.mars5_tts,
281
300
  },
301
+ "mesh-anything": {
302
+ prettyLabel: "MeshAnything",
303
+ repoName: "MeshAnything",
304
+ repoUrl: "https://github.com/buaacyw/MeshAnything",
305
+ filter: false,
306
+ countDownloads: `path:"MeshAnything_350m.pth"`,
307
+ snippets: snippets.mesh_anything,
308
+ },
282
309
  "ml-agents": {
283
310
  prettyLabel: "ml-agents",
284
311
  repoName: "ml-agents",
@@ -304,6 +331,14 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
304
331
  filter: false,
305
332
  countDownloads: `path:"model.safetensors"`,
306
333
  },
334
+ "mlc-llm": {
335
+ prettyLabel: "MLC-LLM",
336
+ repoName: "MLC-LLM",
337
+ repoUrl: "https://github.com/mlc-ai/mlc-llm",
338
+ docsUrl: "https://llm.mlc.ai/docs/",
339
+ filter: false,
340
+ countDownloads: `path:"mlc-chat-config.json"`,
341
+ },
307
342
  nemo: {
308
343
  prettyLabel: "NeMo",
309
344
  repoName: "NeMo",
@@ -4,19 +4,37 @@
4
4
  * Using src/scripts/inference-codegen
5
5
  */
6
6
 
7
- export type FeatureExtractionOutput = unknown[];
7
+ export type FeatureExtractionOutput = Array<number[]>;
8
8
 
9
9
  /**
10
- * Inputs for Text Embedding inference
10
+ * Feature Extraction Input.
11
+ *
12
+ * Auto-generated from TEI specs.
13
+ * For more details, check out
14
+ * https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-tei-import.ts.
11
15
  */
12
16
  export interface FeatureExtractionInput {
13
17
  /**
14
- * The text to get the embeddings of
18
+ * The text to embed.
15
19
  */
16
20
  inputs: string;
21
+ normalize?: boolean;
17
22
  /**
18
- * Additional inference parameters
23
+ * The name of the prompt that should be used by for encoding. If not set, no prompt
24
+ * will be applied.
25
+ *
26
+ * Must be a key in the `Sentence Transformers` configuration `prompts` dictionary.
27
+ *
28
+ * For example if ``prompt_name`` is "query" and the ``prompts`` is {"query": "query: ",
29
+ * ...},
30
+ * then the sentence "What is the capital of France?" will be encoded as
31
+ * "query: What is the capital of France?" because the prompt text will be prepended before
32
+ * any text to encode.
19
33
  */
20
- parameters?: { [key: string]: unknown };
34
+ prompt_name?: string;
35
+ truncate?: boolean;
36
+ truncation_direction?: FeatureExtractionInputTruncationDirection;
21
37
  [property: string]: unknown;
22
38
  }
39
+
40
+ export type FeatureExtractionInputTruncationDirection = "Left" | "Right";
@@ -1,26 +1,47 @@
1
1
  {
2
2
  "$id": "/inference/schemas/feature-extraction/input.json",
3
3
  "$schema": "http://json-schema.org/draft-06/schema#",
4
- "description": "Inputs for Text Embedding inference",
4
+ "description": "Feature Extraction Input.\n\nAuto-generated from TEI specs.\nFor more details, check out https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-tei-import.ts.",
5
5
  "title": "FeatureExtractionInput",
6
6
  "type": "object",
7
+ "required": ["inputs"],
7
8
  "properties": {
8
9
  "inputs": {
9
- "description": "The text to get the embeddings of",
10
- "type": "string"
10
+ "type": "string",
11
+ "description": "The text to embed."
11
12
  },
12
- "parameters": {
13
- "description": "Additional inference parameters",
14
- "$ref": "#/$defs/FeatureExtractionParameters"
13
+ "normalize": {
14
+ "type": "boolean",
15
+ "default": "true",
16
+ "example": "true"
17
+ },
18
+ "prompt_name": {
19
+ "type": "string",
20
+ "description": "The name of the prompt that should be used by for encoding. If not set, no prompt\nwill be applied.\n\nMust be a key in the `Sentence Transformers` configuration `prompts` dictionary.\n\nFor example if ``prompt_name`` is \"query\" and the ``prompts`` is {\"query\": \"query: \", ...},\nthen the sentence \"What is the capital of France?\" will be encoded as\n\"query: What is the capital of France?\" because the prompt text will be prepended before\nany text to encode.",
21
+ "default": "null",
22
+ "example": "null",
23
+ "nullable": true
24
+ },
25
+ "truncate": {
26
+ "type": "boolean",
27
+ "default": "false",
28
+ "example": "false",
29
+ "nullable": true
30
+ },
31
+ "truncation_direction": {
32
+ "allOf": [
33
+ {
34
+ "$ref": "#/$defs/FeatureExtractionInputTruncationDirection"
35
+ }
36
+ ],
37
+ "default": "right"
15
38
  }
16
39
  },
17
40
  "$defs": {
18
- "FeatureExtractionParameters": {
19
- "title": "FeatureExtractionParameters",
20
- "description": "Additional inference parameters for Feature Extraction",
21
- "type": "object",
22
- "properties": {}
41
+ "FeatureExtractionInputTruncationDirection": {
42
+ "type": "string",
43
+ "enum": ["Left", "Right"],
44
+ "title": "FeatureExtractionInputTruncationDirection"
23
45
  }
24
- },
25
- "required": ["inputs"]
46
+ }
26
47
  }
@@ -1,7 +1,15 @@
1
1
  {
2
2
  "$id": "/inference/schemas/feature-extraction/output.json",
3
3
  "$schema": "http://json-schema.org/draft-06/schema#",
4
- "description": "The embedding for the input text, as a nested list (tensor) of floats",
4
+ "description": "Feature Extraction Output.\n\nAuto-generated from TEI specs.\nFor more details, check out https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-tei-import.ts.",
5
+ "title": "FeatureExtractionOutput",
5
6
  "type": "array",
6
- "title": "FeatureExtractionOutput"
7
+ "$defs": {},
8
+ "items": {
9
+ "type": "array",
10
+ "items": {
11
+ "type": "number",
12
+ "format": "float"
13
+ }
14
+ }
7
15
  }
@@ -22,7 +22,7 @@ const taskData: TaskDataCustom = {
22
22
  demo: {
23
23
  inputs: [
24
24
  {
25
- filename: "mask-generation-input.png",
25
+ filename: "image-text-to-text-input.png",
26
26
  type: "img",
27
27
  },
28
28
  {