@huggingface/tasks 0.11.11 → 0.11.12
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 +82 -17
- package/dist/index.js +82 -17
- package/dist/src/dataset-libraries.d.ts +6 -0
- package/dist/src/dataset-libraries.d.ts.map +1 -1
- package/dist/src/hardware.d.ts +8 -0
- package/dist/src/hardware.d.ts.map +1 -1
- package/dist/src/model-libraries-snippets.d.ts +2 -0
- package/dist/src/model-libraries-snippets.d.ts.map +1 -1
- package/dist/src/model-libraries.d.ts +21 -2
- package/dist/src/model-libraries.d.ts.map +1 -1
- package/dist/src/snippets/curl.d.ts.map +1 -1
- package/dist/src/snippets/js.d.ts.map +1 -1
- package/dist/src/snippets/python.d.ts.map +1 -1
- package/package.json +1 -1
- package/src/dataset-libraries.ts +6 -0
- package/src/hardware.ts +8 -0
- package/src/local-apps.ts +1 -1
- package/src/model-libraries-snippets.ts +45 -6
- package/src/model-libraries.ts +19 -0
- package/src/snippets/curl.ts +3 -6
- package/src/snippets/js.ts +1 -2
- package/src/snippets/python.ts +1 -2
package/dist/index.cjs
CHANGED
|
@@ -4461,6 +4461,43 @@ var diffusers = (model) => {
|
|
|
4461
4461
|
return diffusers_default(model);
|
|
4462
4462
|
}
|
|
4463
4463
|
};
|
|
4464
|
+
var cartesia_pytorch = (model) => [
|
|
4465
|
+
`# pip install --no-binary :all: cartesia-pytorch
|
|
4466
|
+
from cartesia_pytorch import ReneLMHeadModel
|
|
4467
|
+
from transformers import AutoTokenizer
|
|
4468
|
+
|
|
4469
|
+
model = ReneLMHeadModel.from_pretrained("${model.id}")
|
|
4470
|
+
tokenizer = AutoTokenizer.from_pretrained("allenai/OLMo-1B-hf")
|
|
4471
|
+
|
|
4472
|
+
in_message = ["Rene Descartes was"]
|
|
4473
|
+
inputs = tokenizer(in_message, return_tensors="pt")
|
|
4474
|
+
|
|
4475
|
+
outputs = model.generate(inputs.input_ids, max_length=50, top_k=100, top_p=0.99)
|
|
4476
|
+
out_message = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
|
|
4477
|
+
|
|
4478
|
+
print(out_message)
|
|
4479
|
+
)`
|
|
4480
|
+
];
|
|
4481
|
+
var cartesia_mlx = (model) => [
|
|
4482
|
+
`import mlx.core as mx
|
|
4483
|
+
import cartesia_mlx as cmx
|
|
4484
|
+
|
|
4485
|
+
model = cmx.from_pretrained("${model.id}")
|
|
4486
|
+
model.set_dtype(mx.float32)
|
|
4487
|
+
|
|
4488
|
+
prompt = "Rene Descartes was"
|
|
4489
|
+
|
|
4490
|
+
for text in model.generate(
|
|
4491
|
+
prompt,
|
|
4492
|
+
max_tokens=500,
|
|
4493
|
+
eval_every_n=5,
|
|
4494
|
+
verbose=True,
|
|
4495
|
+
top_p=0.99,
|
|
4496
|
+
temperature=0.85,
|
|
4497
|
+
):
|
|
4498
|
+
print(text, end="", flush=True)
|
|
4499
|
+
`
|
|
4500
|
+
];
|
|
4464
4501
|
var edsnlp = (model) => {
|
|
4465
4502
|
const packageName = nameWithoutNamespace(model.id).replaceAll("-", "_");
|
|
4466
4503
|
return [
|
|
@@ -4551,12 +4588,12 @@ llm = Llama.from_pretrained(
|
|
|
4551
4588
|
)
|
|
4552
4589
|
|
|
4553
4590
|
llm.create_chat_completion(
|
|
4554
|
-
|
|
4555
|
-
|
|
4556
|
-
|
|
4557
|
-
|
|
4558
|
-
|
|
4559
|
-
|
|
4591
|
+
messages = [
|
|
4592
|
+
{
|
|
4593
|
+
"role": "user",
|
|
4594
|
+
"content": "What is the capital of France?"
|
|
4595
|
+
}
|
|
4596
|
+
]
|
|
4560
4597
|
)`
|
|
4561
4598
|
];
|
|
4562
4599
|
var tf_keras = (model) => [
|
|
@@ -5211,6 +5248,13 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
|
|
|
5211
5248
|
filter: false,
|
|
5212
5249
|
countDownloads: `path:"adapter_config.json"`
|
|
5213
5250
|
},
|
|
5251
|
+
deepforest: {
|
|
5252
|
+
prettyLabel: "DeepForest",
|
|
5253
|
+
repoName: "deepforest",
|
|
5254
|
+
docsUrl: "https://deepforest.readthedocs.io/en/latest/",
|
|
5255
|
+
repoUrl: "https://github.com/weecology/DeepForest",
|
|
5256
|
+
countDownloads: `path_extension:"pt" OR path_extension:"pl"`
|
|
5257
|
+
},
|
|
5214
5258
|
"depth-anything-v2": {
|
|
5215
5259
|
prettyLabel: "DepthAnythingV2",
|
|
5216
5260
|
repoName: "Depth Anything V2",
|
|
@@ -5240,6 +5284,18 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
|
|
|
5240
5284
|
repoName: "doctr",
|
|
5241
5285
|
repoUrl: "https://github.com/mindee/doctr"
|
|
5242
5286
|
},
|
|
5287
|
+
cartesia_pytorch: {
|
|
5288
|
+
prettyLabel: "Cartesia Pytorch",
|
|
5289
|
+
repoName: "Cartesia Pytorch",
|
|
5290
|
+
repoUrl: "https://github.com/cartesia-ai/cartesia_pytorch",
|
|
5291
|
+
snippets: cartesia_pytorch
|
|
5292
|
+
},
|
|
5293
|
+
cartesia_mlx: {
|
|
5294
|
+
prettyLabel: "Cartesia MLX",
|
|
5295
|
+
repoName: "Cartesia MLX",
|
|
5296
|
+
repoUrl: "https://github.com/cartesia-ai/cartesia_mlx",
|
|
5297
|
+
snippets: cartesia_mlx
|
|
5298
|
+
},
|
|
5243
5299
|
edsnlp: {
|
|
5244
5300
|
prettyLabel: "EDS-NLP",
|
|
5245
5301
|
repoName: "edsnlp",
|
|
@@ -5836,8 +5892,7 @@ var snippetBasic = (model, accessToken) => `curl https://api-inference.huggingfa
|
|
|
5836
5892
|
-X POST \\
|
|
5837
5893
|
-d '{"inputs": ${getModelInputSnippet(model, true)}}' \\
|
|
5838
5894
|
-H 'Content-Type: application/json' \\
|
|
5839
|
-
-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}"
|
|
5840
|
-
`;
|
|
5895
|
+
-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}"`;
|
|
5841
5896
|
var snippetTextGeneration = (model, accessToken) => {
|
|
5842
5897
|
if (model.config?.tokenizer_config?.chat_template) {
|
|
5843
5898
|
return `curl 'https://api-inference.huggingface.co/models/${model.id}/v1/chat/completions' \\
|
|
@@ -5858,13 +5913,11 @@ var snippetZeroShotClassification = (model, accessToken) => `curl https://api-in
|
|
|
5858
5913
|
-X POST \\
|
|
5859
5914
|
-d '{"inputs": ${getModelInputSnippet(model, true)}, "parameters": {"candidate_labels": ["refund", "legal", "faq"]}}' \\
|
|
5860
5915
|
-H 'Content-Type: application/json' \\
|
|
5861
|
-
-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}"
|
|
5862
|
-
`;
|
|
5916
|
+
-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}"`;
|
|
5863
5917
|
var snippetFile = (model, accessToken) => `curl https://api-inference.huggingface.co/models/${model.id} \\
|
|
5864
5918
|
-X POST \\
|
|
5865
5919
|
--data-binary '@${getModelInputSnippet(model, true, true)}' \\
|
|
5866
|
-
-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}"
|
|
5867
|
-
`;
|
|
5920
|
+
-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}"`;
|
|
5868
5921
|
var curlSnippets = {
|
|
5869
5922
|
// Same order as in js/src/lib/interfaces/Types.ts
|
|
5870
5923
|
"text-classification": snippetBasic,
|
|
@@ -5925,8 +5978,7 @@ for message in client.chat_completion(
|
|
|
5925
5978
|
max_tokens=500,
|
|
5926
5979
|
stream=True,
|
|
5927
5980
|
):
|
|
5928
|
-
print(message.choices[0].delta.content, end="")
|
|
5929
|
-
`;
|
|
5981
|
+
print(message.choices[0].delta.content, end="")`;
|
|
5930
5982
|
var snippetZeroShotClassification2 = (model) => `def query(payload):
|
|
5931
5983
|
response = requests.post(API_URL, headers=headers, json=payload)
|
|
5932
5984
|
return response.json()
|
|
@@ -6104,8 +6156,7 @@ for await (const chunk of inference.chatCompletionStream({
|
|
|
6104
6156
|
max_tokens: 500,
|
|
6105
6157
|
})) {
|
|
6106
6158
|
process.stdout.write(chunk.choices[0]?.delta?.content || "");
|
|
6107
|
-
}
|
|
6108
|
-
`;
|
|
6159
|
+
}`;
|
|
6109
6160
|
} else {
|
|
6110
6161
|
return snippetBasic3(model, accessToken);
|
|
6111
6162
|
}
|
|
@@ -6370,6 +6421,14 @@ var SKUS = {
|
|
|
6370
6421
|
tflops: 12.74,
|
|
6371
6422
|
memory: [12, 8]
|
|
6372
6423
|
},
|
|
6424
|
+
"RTX 2080 Ti": {
|
|
6425
|
+
tflops: 26.9,
|
|
6426
|
+
memory: [11]
|
|
6427
|
+
},
|
|
6428
|
+
"RTX 2080": {
|
|
6429
|
+
tflops: 20.14,
|
|
6430
|
+
memory: [8]
|
|
6431
|
+
},
|
|
6373
6432
|
"RTX 2070": {
|
|
6374
6433
|
tflops: 14.93,
|
|
6375
6434
|
memory: [8]
|
|
@@ -6789,7 +6848,7 @@ var LOCAL_APPS = {
|
|
|
6789
6848
|
mainTask: "text-to-image",
|
|
6790
6849
|
macOSOnly: true,
|
|
6791
6850
|
displayOnModelPage: (model) => model.library_name === "diffusers" && model.pipeline_tag === "text-to-image",
|
|
6792
|
-
deeplink: (model) => new URL(`diffusionbee
|
|
6851
|
+
deeplink: (model) => new URL(`https://diffusionbee.com/huggingface_import?model_id=${model.id}`)
|
|
6793
6852
|
},
|
|
6794
6853
|
joyfusion: {
|
|
6795
6854
|
prettyLabel: "JoyFusion",
|
|
@@ -6863,6 +6922,12 @@ var DATASET_LIBRARIES_UI_ELEMENTS = {
|
|
|
6863
6922
|
repoName: "polars",
|
|
6864
6923
|
repoUrl: "https://github.com/pola-rs/polars",
|
|
6865
6924
|
docsUrl: "https://huggingface.co/docs/hub/datasets-polars"
|
|
6925
|
+
},
|
|
6926
|
+
duckdb: {
|
|
6927
|
+
prettyLabel: "DuckDB",
|
|
6928
|
+
repoName: "duckdb",
|
|
6929
|
+
repoUrl: "https://github.com/duckdb/duckdb",
|
|
6930
|
+
docsUrl: "https://huggingface.co/docs/hub/datasets-duckdb"
|
|
6866
6931
|
}
|
|
6867
6932
|
};
|
|
6868
6933
|
// Annotate the CommonJS export names for ESM import in node:
|
package/dist/index.js
CHANGED
|
@@ -4423,6 +4423,43 @@ var diffusers = (model) => {
|
|
|
4423
4423
|
return diffusers_default(model);
|
|
4424
4424
|
}
|
|
4425
4425
|
};
|
|
4426
|
+
var cartesia_pytorch = (model) => [
|
|
4427
|
+
`# pip install --no-binary :all: cartesia-pytorch
|
|
4428
|
+
from cartesia_pytorch import ReneLMHeadModel
|
|
4429
|
+
from transformers import AutoTokenizer
|
|
4430
|
+
|
|
4431
|
+
model = ReneLMHeadModel.from_pretrained("${model.id}")
|
|
4432
|
+
tokenizer = AutoTokenizer.from_pretrained("allenai/OLMo-1B-hf")
|
|
4433
|
+
|
|
4434
|
+
in_message = ["Rene Descartes was"]
|
|
4435
|
+
inputs = tokenizer(in_message, return_tensors="pt")
|
|
4436
|
+
|
|
4437
|
+
outputs = model.generate(inputs.input_ids, max_length=50, top_k=100, top_p=0.99)
|
|
4438
|
+
out_message = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
|
|
4439
|
+
|
|
4440
|
+
print(out_message)
|
|
4441
|
+
)`
|
|
4442
|
+
];
|
|
4443
|
+
var cartesia_mlx = (model) => [
|
|
4444
|
+
`import mlx.core as mx
|
|
4445
|
+
import cartesia_mlx as cmx
|
|
4446
|
+
|
|
4447
|
+
model = cmx.from_pretrained("${model.id}")
|
|
4448
|
+
model.set_dtype(mx.float32)
|
|
4449
|
+
|
|
4450
|
+
prompt = "Rene Descartes was"
|
|
4451
|
+
|
|
4452
|
+
for text in model.generate(
|
|
4453
|
+
prompt,
|
|
4454
|
+
max_tokens=500,
|
|
4455
|
+
eval_every_n=5,
|
|
4456
|
+
verbose=True,
|
|
4457
|
+
top_p=0.99,
|
|
4458
|
+
temperature=0.85,
|
|
4459
|
+
):
|
|
4460
|
+
print(text, end="", flush=True)
|
|
4461
|
+
`
|
|
4462
|
+
];
|
|
4426
4463
|
var edsnlp = (model) => {
|
|
4427
4464
|
const packageName = nameWithoutNamespace(model.id).replaceAll("-", "_");
|
|
4428
4465
|
return [
|
|
@@ -4513,12 +4550,12 @@ llm = Llama.from_pretrained(
|
|
|
4513
4550
|
)
|
|
4514
4551
|
|
|
4515
4552
|
llm.create_chat_completion(
|
|
4516
|
-
|
|
4517
|
-
|
|
4518
|
-
|
|
4519
|
-
|
|
4520
|
-
|
|
4521
|
-
|
|
4553
|
+
messages = [
|
|
4554
|
+
{
|
|
4555
|
+
"role": "user",
|
|
4556
|
+
"content": "What is the capital of France?"
|
|
4557
|
+
}
|
|
4558
|
+
]
|
|
4522
4559
|
)`
|
|
4523
4560
|
];
|
|
4524
4561
|
var tf_keras = (model) => [
|
|
@@ -5173,6 +5210,13 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
|
|
|
5173
5210
|
filter: false,
|
|
5174
5211
|
countDownloads: `path:"adapter_config.json"`
|
|
5175
5212
|
},
|
|
5213
|
+
deepforest: {
|
|
5214
|
+
prettyLabel: "DeepForest",
|
|
5215
|
+
repoName: "deepforest",
|
|
5216
|
+
docsUrl: "https://deepforest.readthedocs.io/en/latest/",
|
|
5217
|
+
repoUrl: "https://github.com/weecology/DeepForest",
|
|
5218
|
+
countDownloads: `path_extension:"pt" OR path_extension:"pl"`
|
|
5219
|
+
},
|
|
5176
5220
|
"depth-anything-v2": {
|
|
5177
5221
|
prettyLabel: "DepthAnythingV2",
|
|
5178
5222
|
repoName: "Depth Anything V2",
|
|
@@ -5202,6 +5246,18 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
|
|
|
5202
5246
|
repoName: "doctr",
|
|
5203
5247
|
repoUrl: "https://github.com/mindee/doctr"
|
|
5204
5248
|
},
|
|
5249
|
+
cartesia_pytorch: {
|
|
5250
|
+
prettyLabel: "Cartesia Pytorch",
|
|
5251
|
+
repoName: "Cartesia Pytorch",
|
|
5252
|
+
repoUrl: "https://github.com/cartesia-ai/cartesia_pytorch",
|
|
5253
|
+
snippets: cartesia_pytorch
|
|
5254
|
+
},
|
|
5255
|
+
cartesia_mlx: {
|
|
5256
|
+
prettyLabel: "Cartesia MLX",
|
|
5257
|
+
repoName: "Cartesia MLX",
|
|
5258
|
+
repoUrl: "https://github.com/cartesia-ai/cartesia_mlx",
|
|
5259
|
+
snippets: cartesia_mlx
|
|
5260
|
+
},
|
|
5205
5261
|
edsnlp: {
|
|
5206
5262
|
prettyLabel: "EDS-NLP",
|
|
5207
5263
|
repoName: "edsnlp",
|
|
@@ -5798,8 +5854,7 @@ var snippetBasic = (model, accessToken) => `curl https://api-inference.huggingfa
|
|
|
5798
5854
|
-X POST \\
|
|
5799
5855
|
-d '{"inputs": ${getModelInputSnippet(model, true)}}' \\
|
|
5800
5856
|
-H 'Content-Type: application/json' \\
|
|
5801
|
-
-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}"
|
|
5802
|
-
`;
|
|
5857
|
+
-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}"`;
|
|
5803
5858
|
var snippetTextGeneration = (model, accessToken) => {
|
|
5804
5859
|
if (model.config?.tokenizer_config?.chat_template) {
|
|
5805
5860
|
return `curl 'https://api-inference.huggingface.co/models/${model.id}/v1/chat/completions' \\
|
|
@@ -5820,13 +5875,11 @@ var snippetZeroShotClassification = (model, accessToken) => `curl https://api-in
|
|
|
5820
5875
|
-X POST \\
|
|
5821
5876
|
-d '{"inputs": ${getModelInputSnippet(model, true)}, "parameters": {"candidate_labels": ["refund", "legal", "faq"]}}' \\
|
|
5822
5877
|
-H 'Content-Type: application/json' \\
|
|
5823
|
-
-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}"
|
|
5824
|
-
`;
|
|
5878
|
+
-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}"`;
|
|
5825
5879
|
var snippetFile = (model, accessToken) => `curl https://api-inference.huggingface.co/models/${model.id} \\
|
|
5826
5880
|
-X POST \\
|
|
5827
5881
|
--data-binary '@${getModelInputSnippet(model, true, true)}' \\
|
|
5828
|
-
-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}"
|
|
5829
|
-
`;
|
|
5882
|
+
-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}"`;
|
|
5830
5883
|
var curlSnippets = {
|
|
5831
5884
|
// Same order as in js/src/lib/interfaces/Types.ts
|
|
5832
5885
|
"text-classification": snippetBasic,
|
|
@@ -5887,8 +5940,7 @@ for message in client.chat_completion(
|
|
|
5887
5940
|
max_tokens=500,
|
|
5888
5941
|
stream=True,
|
|
5889
5942
|
):
|
|
5890
|
-
print(message.choices[0].delta.content, end="")
|
|
5891
|
-
`;
|
|
5943
|
+
print(message.choices[0].delta.content, end="")`;
|
|
5892
5944
|
var snippetZeroShotClassification2 = (model) => `def query(payload):
|
|
5893
5945
|
response = requests.post(API_URL, headers=headers, json=payload)
|
|
5894
5946
|
return response.json()
|
|
@@ -6066,8 +6118,7 @@ for await (const chunk of inference.chatCompletionStream({
|
|
|
6066
6118
|
max_tokens: 500,
|
|
6067
6119
|
})) {
|
|
6068
6120
|
process.stdout.write(chunk.choices[0]?.delta?.content || "");
|
|
6069
|
-
}
|
|
6070
|
-
`;
|
|
6121
|
+
}`;
|
|
6071
6122
|
} else {
|
|
6072
6123
|
return snippetBasic3(model, accessToken);
|
|
6073
6124
|
}
|
|
@@ -6332,6 +6383,14 @@ var SKUS = {
|
|
|
6332
6383
|
tflops: 12.74,
|
|
6333
6384
|
memory: [12, 8]
|
|
6334
6385
|
},
|
|
6386
|
+
"RTX 2080 Ti": {
|
|
6387
|
+
tflops: 26.9,
|
|
6388
|
+
memory: [11]
|
|
6389
|
+
},
|
|
6390
|
+
"RTX 2080": {
|
|
6391
|
+
tflops: 20.14,
|
|
6392
|
+
memory: [8]
|
|
6393
|
+
},
|
|
6335
6394
|
"RTX 2070": {
|
|
6336
6395
|
tflops: 14.93,
|
|
6337
6396
|
memory: [8]
|
|
@@ -6751,7 +6810,7 @@ var LOCAL_APPS = {
|
|
|
6751
6810
|
mainTask: "text-to-image",
|
|
6752
6811
|
macOSOnly: true,
|
|
6753
6812
|
displayOnModelPage: (model) => model.library_name === "diffusers" && model.pipeline_tag === "text-to-image",
|
|
6754
|
-
deeplink: (model) => new URL(`diffusionbee
|
|
6813
|
+
deeplink: (model) => new URL(`https://diffusionbee.com/huggingface_import?model_id=${model.id}`)
|
|
6755
6814
|
},
|
|
6756
6815
|
joyfusion: {
|
|
6757
6816
|
prettyLabel: "JoyFusion",
|
|
@@ -6825,6 +6884,12 @@ var DATASET_LIBRARIES_UI_ELEMENTS = {
|
|
|
6825
6884
|
repoName: "polars",
|
|
6826
6885
|
repoUrl: "https://github.com/pola-rs/polars",
|
|
6827
6886
|
docsUrl: "https://huggingface.co/docs/hub/datasets-polars"
|
|
6887
|
+
},
|
|
6888
|
+
duckdb: {
|
|
6889
|
+
prettyLabel: "DuckDB",
|
|
6890
|
+
repoName: "duckdb",
|
|
6891
|
+
repoUrl: "https://github.com/duckdb/duckdb",
|
|
6892
|
+
docsUrl: "https://huggingface.co/docs/hub/datasets-duckdb"
|
|
6828
6893
|
}
|
|
6829
6894
|
};
|
|
6830
6895
|
export {
|
|
@@ -76,6 +76,12 @@ export declare const DATASET_LIBRARIES_UI_ELEMENTS: {
|
|
|
76
76
|
repoUrl: string;
|
|
77
77
|
docsUrl: string;
|
|
78
78
|
};
|
|
79
|
+
duckdb: {
|
|
80
|
+
prettyLabel: string;
|
|
81
|
+
repoName: string;
|
|
82
|
+
repoUrl: string;
|
|
83
|
+
docsUrl: string;
|
|
84
|
+
};
|
|
79
85
|
};
|
|
80
86
|
export type DatasetLibraryKey = keyof typeof DATASET_LIBRARIES_UI_ELEMENTS;
|
|
81
87
|
//# sourceMappingURL=dataset-libraries.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"dataset-libraries.d.ts","sourceRoot":"","sources":["../../src/dataset-libraries.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,WAAW,uBAAuB;IACvC;;;;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;CACjB;AAED,eAAO,MAAM,6BAA6B
|
|
1
|
+
{"version":3,"file":"dataset-libraries.d.ts","sourceRoot":"","sources":["../../src/dataset-libraries.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,WAAW,uBAAuB;IACvC;;;;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;CACjB;AAED,eAAO,MAAM,6BAA6B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA6DS,CAAC;AAGpD,MAAM,MAAM,iBAAiB,GAAG,MAAM,OAAO,6BAA6B,CAAC"}
|
package/dist/src/hardware.d.ts
CHANGED
|
@@ -160,6 +160,14 @@ export declare const SKUS: {
|
|
|
160
160
|
tflops: number;
|
|
161
161
|
memory: number[];
|
|
162
162
|
};
|
|
163
|
+
"RTX 2080 Ti": {
|
|
164
|
+
tflops: number;
|
|
165
|
+
memory: number[];
|
|
166
|
+
};
|
|
167
|
+
"RTX 2080": {
|
|
168
|
+
tflops: number;
|
|
169
|
+
memory: number[];
|
|
170
|
+
};
|
|
163
171
|
"RTX 2070": {
|
|
164
172
|
tflops: number;
|
|
165
173
|
memory: number[];
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"hardware.d.ts","sourceRoot":"","sources":["../../src/hardware.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,eAAO,MAAM,iDAAiD,QAAW,CAAC;AAC1E,eAAO,MAAM,yDAAyD,QAAW,CAAC;AAClF,eAAO,MAAM,oCAAoC,QAAU,CAAC;AAE5D;;;GAGG;AACH,eAAO,MAAM,+CAA+C,QAAW,CAAC;AAExE,MAAM,WAAW,YAAY;IAC5B;;;;;;;;;OASG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;;OAGG;IACH,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;CAClB;AAED,eAAO,MAAM,sBAAsB,UAAqD,CAAC;AAEzF,eAAO,MAAM,IAAI
|
|
1
|
+
{"version":3,"file":"hardware.d.ts","sourceRoot":"","sources":["../../src/hardware.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,eAAO,MAAM,iDAAiD,QAAW,CAAC;AAC1E,eAAO,MAAM,yDAAyD,QAAW,CAAC;AAClF,eAAO,MAAM,oCAAoC,QAAU,CAAC;AAE5D;;;GAGG;AACH,eAAO,MAAM,+CAA+C,QAAW,CAAC;AAExE,MAAM,WAAW,YAAY;IAC5B;;;;;;;;;OASG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;;OAGG;IACH,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;CAClB;AAED,eAAO,MAAM,sBAAsB,UAAqD,CAAC;AAEzF,eAAO,MAAM,IAAI;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA+YuD,CAAC;AAEzE,MAAM,MAAM,OAAO,GAAG,MAAM,OAAO,IAAI,CAAC"}
|
|
@@ -7,6 +7,8 @@ export declare const bertopic: (model: ModelData) => string[];
|
|
|
7
7
|
export declare const bm25s: (model: ModelData) => string[];
|
|
8
8
|
export declare const depth_anything_v2: (model: ModelData) => string[];
|
|
9
9
|
export declare const diffusers: (model: ModelData) => string[];
|
|
10
|
+
export declare const cartesia_pytorch: (model: ModelData) => string[];
|
|
11
|
+
export declare const cartesia_mlx: (model: ModelData) => string[];
|
|
10
12
|
export declare const edsnlp: (model: ModelData) => string[];
|
|
11
13
|
export declare const espnetTTS: (model: ModelData) => string[];
|
|
12
14
|
export declare const espnetASR: (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;AAEF,eAAO,MAAM,iBAAiB,UAAW,SAAS,KAAG,MAAM,EA6C1D,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,gBAAgB,UAAW,SAAS,KAAG,MAAM,EAgBzD,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAOjD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAIlD,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;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AAyBF,eAAO,MAAM,aAAa,UAAW,SAAS,KAAG,MAAM,EAOtD,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAI7C,CAAC;AAEF,eAAO,MAAM,OAAO,QAA6B,MAAM,EAQtD,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAanC,CAAC;AAsCF,eAAO,MAAM,OAAO,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,IAAI,UAAW,SAAS,KAAG,MAAM,EA2B7C,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,QAAQ,UAAW,SAAS,KAAG,MAAM,EAYjD,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;AAEF,eAAO,MAAM,iBAAiB,UAAW,SAAS,KAAG,MAAM,EA6C1D,CAAC;AA+BF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAUlD,CAAC;AAEF,eAAO,MAAM,gBAAgB,UAAW,SAAS,KAAG,MAAM,EAgBzD,CAAC;AAEF,eAAO,MAAM,YAAY,UAAW,SAAS,KAAG,MAAM,EAmBrD,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,gBAAgB,UAAW,SAAS,KAAG,MAAM,EAgBzD,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAOjD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAIlD,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;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AAyBF,eAAO,MAAM,aAAa,UAAW,SAAS,KAAG,MAAM,EAOtD,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAI7C,CAAC;AAEF,eAAO,MAAM,OAAO,QAA6B,MAAM,EAQtD,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAanC,CAAC;AAsCF,eAAO,MAAM,OAAO,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,IAAI,UAAW,SAAS,KAAG,MAAM,EA2B7C,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,QAAQ,UAAW,SAAS,KAAG,MAAM,EAYjD,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"}
|
|
@@ -146,6 +146,13 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
|
|
|
146
146
|
filter: false;
|
|
147
147
|
countDownloads: string;
|
|
148
148
|
};
|
|
149
|
+
deepforest: {
|
|
150
|
+
prettyLabel: string;
|
|
151
|
+
repoName: string;
|
|
152
|
+
docsUrl: string;
|
|
153
|
+
repoUrl: string;
|
|
154
|
+
countDownloads: string;
|
|
155
|
+
};
|
|
149
156
|
"depth-anything-v2": {
|
|
150
157
|
prettyLabel: string;
|
|
151
158
|
repoName: string;
|
|
@@ -174,6 +181,18 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
|
|
|
174
181
|
repoName: string;
|
|
175
182
|
repoUrl: string;
|
|
176
183
|
};
|
|
184
|
+
cartesia_pytorch: {
|
|
185
|
+
prettyLabel: string;
|
|
186
|
+
repoName: string;
|
|
187
|
+
repoUrl: string;
|
|
188
|
+
snippets: (model: ModelData) => string[];
|
|
189
|
+
};
|
|
190
|
+
cartesia_mlx: {
|
|
191
|
+
prettyLabel: string;
|
|
192
|
+
repoName: string;
|
|
193
|
+
repoUrl: string;
|
|
194
|
+
snippets: (model: ModelData) => string[];
|
|
195
|
+
};
|
|
177
196
|
edsnlp: {
|
|
178
197
|
prettyLabel: string;
|
|
179
198
|
repoName: string;
|
|
@@ -629,6 +648,6 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
|
|
|
629
648
|
};
|
|
630
649
|
};
|
|
631
650
|
export type ModelLibraryKey = keyof typeof MODEL_LIBRARIES_UI_ELEMENTS;
|
|
632
|
-
export declare const ALL_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "audioseal" | "bertopic" | "big_vision" | "birefnet" | "bm25s" | "champ" | "chat_tts" | "colpali" | "depth-anything-v2" | "diffree" | "diffusers" | "doctr" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gemma.cpp" | "gliner" | "glyph-byt5" | "grok" | "hallo" | "hunyuan-dit" | "keras" | "tf-keras" | "keras-nlp" | "k2" | "liveportrait" | "llama-cpp-python" | "mindspore" | "mamba-ssm" | "mars5-tts" | "mesh-anything" | "ml-agents" | "mlx" | "mlx-image" | "mlc-llm" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "recurrentgemma" | "relik" | "refiners" | "saelens" | "sam2" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "stable-audio-tools" | "diffusion-single-file" | "seed-story" | "stable-baselines3" | "stanza" | "tensorflowtts" | "tic-clip" | "timesfm" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "voicecraft" | "whisperkit")[];
|
|
633
|
-
export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "audioseal" | "bertopic" | "big_vision" | "birefnet" | "bm25s" | "champ" | "chat_tts" | "colpali" | "depth-anything-v2" | "diffree" | "diffusers" | "doctr" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gemma.cpp" | "gliner" | "glyph-byt5" | "grok" | "hallo" | "hunyuan-dit" | "keras" | "tf-keras" | "keras-nlp" | "k2" | "liveportrait" | "llama-cpp-python" | "mindspore" | "mamba-ssm" | "mars5-tts" | "mesh-anything" | "ml-agents" | "mlx" | "mlx-image" | "mlc-llm" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "recurrentgemma" | "relik" | "refiners" | "saelens" | "sam2" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "stable-audio-tools" | "diffusion-single-file" | "seed-story" | "stable-baselines3" | "stanza" | "tensorflowtts" | "tic-clip" | "timesfm" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "voicecraft" | "whisperkit")[];
|
|
651
|
+
export declare const ALL_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "audioseal" | "bertopic" | "big_vision" | "birefnet" | "bm25s" | "champ" | "chat_tts" | "colpali" | "deepforest" | "depth-anything-v2" | "diffree" | "diffusers" | "doctr" | "cartesia_pytorch" | "cartesia_mlx" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gemma.cpp" | "gliner" | "glyph-byt5" | "grok" | "hallo" | "hunyuan-dit" | "keras" | "tf-keras" | "keras-nlp" | "k2" | "liveportrait" | "llama-cpp-python" | "mindspore" | "mamba-ssm" | "mars5-tts" | "mesh-anything" | "ml-agents" | "mlx" | "mlx-image" | "mlc-llm" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "recurrentgemma" | "relik" | "refiners" | "saelens" | "sam2" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "stable-audio-tools" | "diffusion-single-file" | "seed-story" | "stable-baselines3" | "stanza" | "tensorflowtts" | "tic-clip" | "timesfm" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "voicecraft" | "whisperkit")[];
|
|
652
|
+
export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "audioseal" | "bertopic" | "big_vision" | "birefnet" | "bm25s" | "champ" | "chat_tts" | "colpali" | "deepforest" | "depth-anything-v2" | "diffree" | "diffusers" | "doctr" | "cartesia_pytorch" | "cartesia_mlx" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gemma.cpp" | "gliner" | "glyph-byt5" | "grok" | "hallo" | "hunyuan-dit" | "keras" | "tf-keras" | "keras-nlp" | "k2" | "liveportrait" | "llama-cpp-python" | "mindspore" | "mamba-ssm" | "mars5-tts" | "mesh-anything" | "ml-agents" | "mlx" | "mlx-image" | "mlc-llm" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "recurrentgemma" | "relik" | "refiners" | "saelens" | "sam2" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "stable-audio-tools" | "diffusion-single-file" | "seed-story" | "stable-baselines3" | "stanza" | "tensorflowtts" | "tic-clip" | "timesfm" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "voicecraft" | "whisperkit")[];
|
|
634
653
|
//# 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;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAmlBI,CAAC;AAE7C,MAAM,MAAM,eAAe,GAAG,MAAM,OAAO,2BAA2B,CAAC;AAEvE,eAAO,MAAM,sBAAsB,+kCAAgE,CAAC;AAEpG,eAAO,MAAM,8BAA8B,+kCAQ1B,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"curl.d.ts","sourceRoot":"","sources":["../../../src/snippets/curl.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAEpD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAEnD,eAAO,MAAM,YAAY,UAAW,gBAAgB,eAAe,MAAM,KAAG,
|
|
1
|
+
{"version":3,"file":"curl.d.ts","sourceRoot":"","sources":["../../../src/snippets/curl.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAEpD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAEnD,eAAO,MAAM,YAAY,UAAW,gBAAgB,eAAe,MAAM,KAAG,MAKhB,CAAC;AAE7D,eAAO,MAAM,qBAAqB,UAAW,gBAAgB,eAAe,MAAM,KAAG,MAgBpF,CAAC;AAEF,eAAO,MAAM,6BAA6B,UAAW,gBAAgB,eAAe,MAAM,KAAG,MAKjC,CAAC;AAE7D,eAAO,MAAM,WAAW,UAAW,gBAAgB,eAAe,MAAM,KAAG,MAIf,CAAC;AAE7D,eAAO,MAAM,YAAY,EAAE,OAAO,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,KAAK,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,KAAK,MAAM,CAAC,CAwBhH,CAAC;AAEF,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,GAAG,MAAM,CAI5F;AAED,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,IAAI,CAAC,gBAAgB,EAAE,cAAc,CAAC,GAAG,OAAO,CAE9F"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"js.d.ts","sourceRoot":"","sources":["../../../src/snippets/js.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAEpD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAEnD,eAAO,MAAM,YAAY,UAAW,gBAAgB,eAAe,MAAM,KAAG,MAmBxE,CAAC;AAEL,eAAO,MAAM,qBAAqB,UAAW,gBAAgB,eAAe,MAAM,KAAG,
|
|
1
|
+
{"version":3,"file":"js.d.ts","sourceRoot":"","sources":["../../../src/snippets/js.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAEpD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAEnD,eAAO,MAAM,YAAY,UAAW,gBAAgB,eAAe,MAAM,KAAG,MAmBxE,CAAC;AAEL,eAAO,MAAM,qBAAqB,UAAW,gBAAgB,eAAe,MAAM,KAAG,MAiBpF,CAAC;AACF,eAAO,MAAM,6BAA6B,UAAW,gBAAgB,eAAe,MAAM,KAAG,MAqBzF,CAAC;AAEL,eAAO,MAAM,kBAAkB,UAAW,gBAAgB,eAAe,MAAM,KAAG,MAkB9E,CAAC;AAEL,eAAO,MAAM,kBAAkB,UAAW,gBAAgB,eAAe,MAAM,KAAG,MAqCjF,CAAC;AAEF,eAAO,MAAM,WAAW,UAAW,gBAAgB,eAAe,MAAM,KAAG,MAoBvE,CAAC;AAEL,eAAO,MAAM,UAAU,EAAE,OAAO,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,KAAK,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,KAAK,MAAM,CAAC,CAwB9G,CAAC;AAEF,wBAAgB,qBAAqB,CAAC,KAAK,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,GAAG,MAAM,CAI1F;AAED,wBAAgB,qBAAqB,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAEtE"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"python.d.ts","sourceRoot":"","sources":["../../../src/snippets/python.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAEpD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAEnD,eAAO,MAAM,qBAAqB,UAAW,gBAAgB,eAAe,MAAM,KAAG,
|
|
1
|
+
{"version":3,"file":"python.d.ts","sourceRoot":"","sources":["../../../src/snippets/python.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAEpD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAEnD,eAAO,MAAM,qBAAqB,UAAW,gBAAgB,eAAe,MAAM,KAAG,MAajC,CAAC;AAErD,eAAO,MAAM,6BAA6B,UAAW,gBAAgB,KAAG,MAQrE,CAAC;AAEJ,eAAO,MAAM,kCAAkC,UAAW,gBAAgB,KAAG,MAc1E,CAAC;AAEJ,eAAO,MAAM,YAAY,UAAW,gBAAgB,KAAG,MAOpD,CAAC;AAEJ,eAAO,MAAM,WAAW,UAAW,gBAAgB,KAAG,MAOP,CAAC;AAEhD,eAAO,MAAM,kBAAkB,UAAW,gBAAgB,KAAG,MAUjB,CAAC;AAE7C,eAAO,MAAM,cAAc,UAAW,gBAAgB,KAAG,MAMtD,CAAC;AAEJ,eAAO,MAAM,kBAAkB,UAAW,gBAAgB,KAAG,MA2B5D,CAAC;AAEF,eAAO,MAAM,gCAAgC,UAAW,gBAAgB,KAAG,MAUxE,CAAC;AAEJ,eAAO,MAAM,cAAc,EAAE,OAAO,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,KAAK,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,KAAK,MAAM,CAAC,CA4BlH,CAAC;AAEF,wBAAgB,yBAAyB,CAAC,KAAK,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,GAAG,MAAM,CAiB9F;AAED,wBAAgB,yBAAyB,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAE1E"}
|
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.11.
|
|
4
|
+
"version": "0.11.12",
|
|
5
5
|
"description": "List of ML tasks for huggingface.co/tasks",
|
|
6
6
|
"repository": "https://github.com/huggingface/huggingface.js.git",
|
|
7
7
|
"publishConfig": {
|
package/src/dataset-libraries.ts
CHANGED
|
@@ -77,6 +77,12 @@ export const DATASET_LIBRARIES_UI_ELEMENTS = {
|
|
|
77
77
|
repoUrl: "https://github.com/pola-rs/polars",
|
|
78
78
|
docsUrl: "https://huggingface.co/docs/hub/datasets-polars",
|
|
79
79
|
},
|
|
80
|
+
duckdb: {
|
|
81
|
+
prettyLabel: "DuckDB",
|
|
82
|
+
repoName: "duckdb",
|
|
83
|
+
repoUrl: "https://github.com/duckdb/duckdb",
|
|
84
|
+
docsUrl: "https://huggingface.co/docs/hub/datasets-duckdb",
|
|
85
|
+
},
|
|
80
86
|
} satisfies Record<string, DatasetLibraryUiElement>;
|
|
81
87
|
|
|
82
88
|
/// List of the dataset libraries supported by the Hub
|
package/src/hardware.ts
CHANGED
|
@@ -164,6 +164,14 @@ export const SKUS = {
|
|
|
164
164
|
tflops: 12.74,
|
|
165
165
|
memory: [12, 8],
|
|
166
166
|
},
|
|
167
|
+
"RTX 2080 Ti": {
|
|
168
|
+
tflops: 26.9,
|
|
169
|
+
memory: [11],
|
|
170
|
+
},
|
|
171
|
+
"RTX 2080": {
|
|
172
|
+
tflops: 20.14,
|
|
173
|
+
memory: [8],
|
|
174
|
+
},
|
|
167
175
|
"RTX 2070": {
|
|
168
176
|
tflops: 14.93,
|
|
169
177
|
memory: [8],
|
package/src/local-apps.ts
CHANGED
|
@@ -237,7 +237,7 @@ export const LOCAL_APPS = {
|
|
|
237
237
|
mainTask: "text-to-image",
|
|
238
238
|
macOSOnly: true,
|
|
239
239
|
displayOnModelPage: (model) => model.library_name === "diffusers" && model.pipeline_tag === "text-to-image",
|
|
240
|
-
deeplink: (model) => new URL(`diffusionbee
|
|
240
|
+
deeplink: (model) => new URL(`https://diffusionbee.com/huggingface_import?model_id=${model.id}`),
|
|
241
241
|
},
|
|
242
242
|
joyfusion: {
|
|
243
243
|
prettyLabel: "JoyFusion",
|
|
@@ -170,6 +170,45 @@ export const diffusers = (model: ModelData): string[] => {
|
|
|
170
170
|
}
|
|
171
171
|
};
|
|
172
172
|
|
|
173
|
+
export const cartesia_pytorch = (model: ModelData): string[] => [
|
|
174
|
+
`# pip install --no-binary :all: cartesia-pytorch
|
|
175
|
+
from cartesia_pytorch import ReneLMHeadModel
|
|
176
|
+
from transformers import AutoTokenizer
|
|
177
|
+
|
|
178
|
+
model = ReneLMHeadModel.from_pretrained("${model.id}")
|
|
179
|
+
tokenizer = AutoTokenizer.from_pretrained("allenai/OLMo-1B-hf")
|
|
180
|
+
|
|
181
|
+
in_message = ["Rene Descartes was"]
|
|
182
|
+
inputs = tokenizer(in_message, return_tensors="pt")
|
|
183
|
+
|
|
184
|
+
outputs = model.generate(inputs.input_ids, max_length=50, top_k=100, top_p=0.99)
|
|
185
|
+
out_message = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
|
|
186
|
+
|
|
187
|
+
print(out_message)
|
|
188
|
+
)`,
|
|
189
|
+
];
|
|
190
|
+
|
|
191
|
+
export const cartesia_mlx = (model: ModelData): string[] => [
|
|
192
|
+
`import mlx.core as mx
|
|
193
|
+
import cartesia_mlx as cmx
|
|
194
|
+
|
|
195
|
+
model = cmx.from_pretrained("${model.id}")
|
|
196
|
+
model.set_dtype(mx.float32)
|
|
197
|
+
|
|
198
|
+
prompt = "Rene Descartes was"
|
|
199
|
+
|
|
200
|
+
for text in model.generate(
|
|
201
|
+
prompt,
|
|
202
|
+
max_tokens=500,
|
|
203
|
+
eval_every_n=5,
|
|
204
|
+
verbose=True,
|
|
205
|
+
top_p=0.99,
|
|
206
|
+
temperature=0.85,
|
|
207
|
+
):
|
|
208
|
+
print(text, end="", flush=True)
|
|
209
|
+
`,
|
|
210
|
+
];
|
|
211
|
+
|
|
173
212
|
export const edsnlp = (model: ModelData): string[] => {
|
|
174
213
|
const packageName = nameWithoutNamespace(model.id).replaceAll("-", "_");
|
|
175
214
|
return [
|
|
@@ -270,12 +309,12 @@ llm = Llama.from_pretrained(
|
|
|
270
309
|
)
|
|
271
310
|
|
|
272
311
|
llm.create_chat_completion(
|
|
273
|
-
|
|
274
|
-
|
|
275
|
-
|
|
276
|
-
|
|
277
|
-
|
|
278
|
-
|
|
312
|
+
messages = [
|
|
313
|
+
{
|
|
314
|
+
"role": "user",
|
|
315
|
+
"content": "What is the capital of France?"
|
|
316
|
+
}
|
|
317
|
+
]
|
|
279
318
|
)`,
|
|
280
319
|
];
|
|
281
320
|
|
package/src/model-libraries.ts
CHANGED
|
@@ -150,6 +150,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
|
|
|
150
150
|
filter: false,
|
|
151
151
|
countDownloads: `path:"adapter_config.json"`,
|
|
152
152
|
},
|
|
153
|
+
deepforest: {
|
|
154
|
+
prettyLabel: "DeepForest",
|
|
155
|
+
repoName: "deepforest",
|
|
156
|
+
docsUrl: "https://deepforest.readthedocs.io/en/latest/",
|
|
157
|
+
repoUrl: "https://github.com/weecology/DeepForest",
|
|
158
|
+
countDownloads: `path_extension:"pt" OR path_extension:"pl"`,
|
|
159
|
+
},
|
|
153
160
|
"depth-anything-v2": {
|
|
154
161
|
prettyLabel: "DepthAnythingV2",
|
|
155
162
|
repoName: "Depth Anything V2",
|
|
@@ -179,6 +186,18 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
|
|
|
179
186
|
repoName: "doctr",
|
|
180
187
|
repoUrl: "https://github.com/mindee/doctr",
|
|
181
188
|
},
|
|
189
|
+
cartesia_pytorch: {
|
|
190
|
+
prettyLabel: "Cartesia Pytorch",
|
|
191
|
+
repoName: "Cartesia Pytorch",
|
|
192
|
+
repoUrl: "https://github.com/cartesia-ai/cartesia_pytorch",
|
|
193
|
+
snippets: snippets.cartesia_pytorch,
|
|
194
|
+
},
|
|
195
|
+
cartesia_mlx: {
|
|
196
|
+
prettyLabel: "Cartesia MLX",
|
|
197
|
+
repoName: "Cartesia MLX",
|
|
198
|
+
repoUrl: "https://github.com/cartesia-ai/cartesia_mlx",
|
|
199
|
+
snippets: snippets.cartesia_mlx,
|
|
200
|
+
},
|
|
182
201
|
edsnlp: {
|
|
183
202
|
prettyLabel: "EDS-NLP",
|
|
184
203
|
repoName: "edsnlp",
|
package/src/snippets/curl.ts
CHANGED
|
@@ -7,8 +7,7 @@ export const snippetBasic = (model: ModelDataMinimal, accessToken: string): stri
|
|
|
7
7
|
-X POST \\
|
|
8
8
|
-d '{"inputs": ${getModelInputSnippet(model, true)}}' \\
|
|
9
9
|
-H 'Content-Type: application/json' \\
|
|
10
|
-
-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}"
|
|
11
|
-
`;
|
|
10
|
+
-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}"`;
|
|
12
11
|
|
|
13
12
|
export const snippetTextGeneration = (model: ModelDataMinimal, accessToken: string): string => {
|
|
14
13
|
if (model.config?.tokenizer_config?.chat_template) {
|
|
@@ -33,15 +32,13 @@ export const snippetZeroShotClassification = (model: ModelDataMinimal, accessTok
|
|
|
33
32
|
-X POST \\
|
|
34
33
|
-d '{"inputs": ${getModelInputSnippet(model, true)}, "parameters": {"candidate_labels": ["refund", "legal", "faq"]}}' \\
|
|
35
34
|
-H 'Content-Type: application/json' \\
|
|
36
|
-
-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}"
|
|
37
|
-
`;
|
|
35
|
+
-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}"`;
|
|
38
36
|
|
|
39
37
|
export const snippetFile = (model: ModelDataMinimal, accessToken: string): string =>
|
|
40
38
|
`curl https://api-inference.huggingface.co/models/${model.id} \\
|
|
41
39
|
-X POST \\
|
|
42
40
|
--data-binary '@${getModelInputSnippet(model, true, true)}' \\
|
|
43
|
-
-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}"
|
|
44
|
-
`;
|
|
41
|
+
-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}"`;
|
|
45
42
|
|
|
46
43
|
export const curlSnippets: Partial<Record<PipelineType, (model: ModelDataMinimal, accessToken: string) => string>> = {
|
|
47
44
|
// Same order as in js/src/lib/interfaces/Types.ts
|
package/src/snippets/js.ts
CHANGED
package/src/snippets/python.ts
CHANGED
|
@@ -15,8 +15,7 @@ for message in client.chat_completion(
|
|
|
15
15
|
max_tokens=500,
|
|
16
16
|
stream=True,
|
|
17
17
|
):
|
|
18
|
-
print(message.choices[0].delta.content, end="")
|
|
19
|
-
`;
|
|
18
|
+
print(message.choices[0].delta.content, end="")`;
|
|
20
19
|
|
|
21
20
|
export const snippetZeroShotClassification = (model: ModelDataMinimal): string =>
|
|
22
21
|
`def query(payload):
|