@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 +45 -14
- package/dist/index.js +45 -13
- package/dist/scripts/inference-tei-import.d.ts +2 -0
- package/dist/scripts/inference-tei-import.d.ts.map +1 -0
- package/dist/src/index.d.ts +0 -1
- package/dist/src/index.d.ts.map +1 -1
- package/dist/src/model-data.d.ts +2 -15
- package/dist/src/model-data.d.ts.map +1 -1
- package/dist/src/model-libraries-snippets.d.ts +1 -0
- package/dist/src/model-libraries-snippets.d.ts.map +1 -1
- package/dist/src/model-libraries.d.ts +37 -2
- package/dist/src/model-libraries.d.ts.map +1 -1
- package/dist/src/tasks/feature-extraction/inference.d.ts +22 -7
- package/dist/src/tasks/feature-extraction/inference.d.ts.map +1 -1
- package/package.json +3 -2
- package/src/index.ts +0 -1
- package/src/model-data.ts +2 -16
- package/src/model-libraries-snippets.ts +10 -0
- package/src/model-libraries.ts +35 -0
- package/src/tasks/feature-extraction/inference.ts +23 -5
- package/src/tasks/feature-extraction/spec/input.json +34 -13
- package/src/tasks/feature-extraction/spec/output.json +10 -2
- package/src/tasks/image-text-to-text/data.ts +1 -1
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: "
|
|
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: "
|
|
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 @@
|
|
|
1
|
+
{"version":3,"file":"inference-tei-import.d.ts","sourceRoot":"","sources":["../../scripts/inference-tei-import.ts"],"names":[],"mappings":""}
|
package/dist/src/index.d.ts
CHANGED
|
@@ -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 };
|
package/dist/src/index.d.ts.map
CHANGED
|
@@ -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,
|
|
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"}
|
package/dist/src/model-data.d.ts
CHANGED
|
@@ -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:
|
|
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
|
|
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
|
|
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 =
|
|
6
|
+
export type FeatureExtractionOutput = Array<number[]>;
|
|
7
7
|
/**
|
|
8
|
-
*
|
|
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
|
|
16
|
+
* The text to embed.
|
|
13
17
|
*/
|
|
14
18
|
inputs: string;
|
|
19
|
+
normalize?: boolean;
|
|
15
20
|
/**
|
|
16
|
-
*
|
|
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
|
-
|
|
19
|
-
|
|
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,
|
|
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.
|
|
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:
|
|
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
|
|
package/src/model-libraries.ts
CHANGED
|
@@ -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 =
|
|
7
|
+
export type FeatureExtractionOutput = Array<number[]>;
|
|
8
8
|
|
|
9
9
|
/**
|
|
10
|
-
*
|
|
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
|
|
18
|
+
* The text to embed.
|
|
15
19
|
*/
|
|
16
20
|
inputs: string;
|
|
21
|
+
normalize?: boolean;
|
|
17
22
|
/**
|
|
18
|
-
*
|
|
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
|
-
|
|
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": "
|
|
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
|
-
"
|
|
10
|
-
"
|
|
10
|
+
"type": "string",
|
|
11
|
+
"description": "The text to embed."
|
|
11
12
|
},
|
|
12
|
-
"
|
|
13
|
-
"
|
|
14
|
-
"
|
|
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
|
-
"
|
|
19
|
-
"
|
|
20
|
-
"
|
|
21
|
-
"
|
|
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": "
|
|
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
|
-
"
|
|
7
|
+
"$defs": {},
|
|
8
|
+
"items": {
|
|
9
|
+
"type": "array",
|
|
10
|
+
"items": {
|
|
11
|
+
"type": "number",
|
|
12
|
+
"format": "float"
|
|
13
|
+
}
|
|
14
|
+
}
|
|
7
15
|
}
|