@huggingface/tasks 0.10.6 → 0.10.8
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 +35 -6
- package/dist/index.js +35 -6
- package/dist/src/hardware.d.ts +29 -1
- package/dist/src/hardware.d.ts.map +1 -1
- package/dist/src/library-to-tasks.d.ts.map +1 -1
- package/dist/src/model-libraries-snippets.d.ts.map +1 -1
- package/dist/src/model-libraries.d.ts +2 -2
- package/package.json +1 -1
- package/src/hardware.ts +32 -4
- package/src/library-to-tasks.ts +1 -0
- package/src/model-libraries-snippets.ts +2 -1
package/dist/index.cjs
CHANGED
|
@@ -104,7 +104,8 @@ var LIBRARY_TASK_MAPPING = {
|
|
|
104
104
|
"video-classification",
|
|
105
105
|
"visual-question-answering",
|
|
106
106
|
"zero-shot-classification",
|
|
107
|
-
"zero-shot-image-classification"
|
|
107
|
+
"zero-shot-image-classification",
|
|
108
|
+
"zero-shot-object-detection"
|
|
108
109
|
],
|
|
109
110
|
mindspore: ["image-classification"]
|
|
110
111
|
};
|
|
@@ -4424,7 +4425,7 @@ var transformers = (model) => {
|
|
|
4424
4425
|
`model = ${info.auto_model}.from_pretrained("${model.id}"` + remote_code_snippet + ")"
|
|
4425
4426
|
].join("\n");
|
|
4426
4427
|
}
|
|
4427
|
-
if (model.pipeline_tag) {
|
|
4428
|
+
if (model.pipeline_tag && LIBRARY_TASK_MAPPING.transformers?.includes(model.pipeline_tag)) {
|
|
4428
4429
|
const pipelineSnippet = [
|
|
4429
4430
|
"# Use a pipeline as a high-level helper",
|
|
4430
4431
|
"from transformers import pipeline",
|
|
@@ -5459,7 +5460,7 @@ var TFLOPS_THRESHOLD_WHITE_HOUSE_MODEL_TRAINING_TOTAL = 10 ** 14;
|
|
|
5459
5460
|
var TFLOPS_THRESHOLD_WHITE_HOUSE_MODEL_TRAINING_TOTAL_BIOLOGY = 10 ** 11;
|
|
5460
5461
|
var TFLOPS_THRESHOLD_WHITE_HOUSE_CLUSTER = 10 ** 8;
|
|
5461
5462
|
var TFLOPS_THRESHOLD_EU_AI_ACT_MODEL_TRAINING_TOTAL = 10 ** 13;
|
|
5462
|
-
var DEFAULT_MEMORY_OPTIONS = [8, 16, 24, 32, 40, 48, 64, 80, 96, 128];
|
|
5463
|
+
var DEFAULT_MEMORY_OPTIONS = [8, 16, 24, 32, 40, 48, 64, 80, 96, 128, 256, 512];
|
|
5463
5464
|
var SKUS = {
|
|
5464
5465
|
GPU: {
|
|
5465
5466
|
NVIDIA: {
|
|
@@ -5527,13 +5528,41 @@ var SKUS = {
|
|
|
5527
5528
|
tflops: 40,
|
|
5528
5529
|
memory: [24]
|
|
5529
5530
|
},
|
|
5531
|
+
"RTX 3080": {
|
|
5532
|
+
tflops: 30.6,
|
|
5533
|
+
memory: [12, 10]
|
|
5534
|
+
},
|
|
5530
5535
|
"RTX 3080 Ti": {
|
|
5531
5536
|
tflops: 34.1,
|
|
5532
5537
|
memory: [12]
|
|
5533
5538
|
},
|
|
5534
|
-
"RTX
|
|
5535
|
-
tflops:
|
|
5536
|
-
memory: [
|
|
5539
|
+
"RTX 3070": {
|
|
5540
|
+
tflops: 20.31,
|
|
5541
|
+
memory: [8]
|
|
5542
|
+
},
|
|
5543
|
+
"RTX 3070 Ti": {
|
|
5544
|
+
tflops: 21.75,
|
|
5545
|
+
memory: [8]
|
|
5546
|
+
},
|
|
5547
|
+
"RTX 3070 Ti Laptop": {
|
|
5548
|
+
tflops: 16.6,
|
|
5549
|
+
memory: [8]
|
|
5550
|
+
},
|
|
5551
|
+
"RTX 3060 Ti": {
|
|
5552
|
+
tflops: 16.2,
|
|
5553
|
+
memory: [8]
|
|
5554
|
+
},
|
|
5555
|
+
"RTX 3060": {
|
|
5556
|
+
tflops: 12.74,
|
|
5557
|
+
memory: [12, 8]
|
|
5558
|
+
},
|
|
5559
|
+
"RTX 2060 Mobile": {
|
|
5560
|
+
tflops: 9.22,
|
|
5561
|
+
memory: [6]
|
|
5562
|
+
},
|
|
5563
|
+
"RTX Titan": {
|
|
5564
|
+
tflops: 32.62,
|
|
5565
|
+
memory: [24]
|
|
5537
5566
|
}
|
|
5538
5567
|
},
|
|
5539
5568
|
AMD: {
|
package/dist/index.js
CHANGED
|
@@ -66,7 +66,8 @@ var LIBRARY_TASK_MAPPING = {
|
|
|
66
66
|
"video-classification",
|
|
67
67
|
"visual-question-answering",
|
|
68
68
|
"zero-shot-classification",
|
|
69
|
-
"zero-shot-image-classification"
|
|
69
|
+
"zero-shot-image-classification",
|
|
70
|
+
"zero-shot-object-detection"
|
|
70
71
|
],
|
|
71
72
|
mindspore: ["image-classification"]
|
|
72
73
|
};
|
|
@@ -4386,7 +4387,7 @@ var transformers = (model) => {
|
|
|
4386
4387
|
`model = ${info.auto_model}.from_pretrained("${model.id}"` + remote_code_snippet + ")"
|
|
4387
4388
|
].join("\n");
|
|
4388
4389
|
}
|
|
4389
|
-
if (model.pipeline_tag) {
|
|
4390
|
+
if (model.pipeline_tag && LIBRARY_TASK_MAPPING.transformers?.includes(model.pipeline_tag)) {
|
|
4390
4391
|
const pipelineSnippet = [
|
|
4391
4392
|
"# Use a pipeline as a high-level helper",
|
|
4392
4393
|
"from transformers import pipeline",
|
|
@@ -5421,7 +5422,7 @@ var TFLOPS_THRESHOLD_WHITE_HOUSE_MODEL_TRAINING_TOTAL = 10 ** 14;
|
|
|
5421
5422
|
var TFLOPS_THRESHOLD_WHITE_HOUSE_MODEL_TRAINING_TOTAL_BIOLOGY = 10 ** 11;
|
|
5422
5423
|
var TFLOPS_THRESHOLD_WHITE_HOUSE_CLUSTER = 10 ** 8;
|
|
5423
5424
|
var TFLOPS_THRESHOLD_EU_AI_ACT_MODEL_TRAINING_TOTAL = 10 ** 13;
|
|
5424
|
-
var DEFAULT_MEMORY_OPTIONS = [8, 16, 24, 32, 40, 48, 64, 80, 96, 128];
|
|
5425
|
+
var DEFAULT_MEMORY_OPTIONS = [8, 16, 24, 32, 40, 48, 64, 80, 96, 128, 256, 512];
|
|
5425
5426
|
var SKUS = {
|
|
5426
5427
|
GPU: {
|
|
5427
5428
|
NVIDIA: {
|
|
@@ -5489,13 +5490,41 @@ var SKUS = {
|
|
|
5489
5490
|
tflops: 40,
|
|
5490
5491
|
memory: [24]
|
|
5491
5492
|
},
|
|
5493
|
+
"RTX 3080": {
|
|
5494
|
+
tflops: 30.6,
|
|
5495
|
+
memory: [12, 10]
|
|
5496
|
+
},
|
|
5492
5497
|
"RTX 3080 Ti": {
|
|
5493
5498
|
tflops: 34.1,
|
|
5494
5499
|
memory: [12]
|
|
5495
5500
|
},
|
|
5496
|
-
"RTX
|
|
5497
|
-
tflops:
|
|
5498
|
-
memory: [
|
|
5501
|
+
"RTX 3070": {
|
|
5502
|
+
tflops: 20.31,
|
|
5503
|
+
memory: [8]
|
|
5504
|
+
},
|
|
5505
|
+
"RTX 3070 Ti": {
|
|
5506
|
+
tflops: 21.75,
|
|
5507
|
+
memory: [8]
|
|
5508
|
+
},
|
|
5509
|
+
"RTX 3070 Ti Laptop": {
|
|
5510
|
+
tflops: 16.6,
|
|
5511
|
+
memory: [8]
|
|
5512
|
+
},
|
|
5513
|
+
"RTX 3060 Ti": {
|
|
5514
|
+
tflops: 16.2,
|
|
5515
|
+
memory: [8]
|
|
5516
|
+
},
|
|
5517
|
+
"RTX 3060": {
|
|
5518
|
+
tflops: 12.74,
|
|
5519
|
+
memory: [12, 8]
|
|
5520
|
+
},
|
|
5521
|
+
"RTX 2060 Mobile": {
|
|
5522
|
+
tflops: 9.22,
|
|
5523
|
+
memory: [6]
|
|
5524
|
+
},
|
|
5525
|
+
"RTX Titan": {
|
|
5526
|
+
tflops: 32.62,
|
|
5527
|
+
memory: [24]
|
|
5499
5528
|
}
|
|
5500
5529
|
},
|
|
5501
5530
|
AMD: {
|
package/dist/src/hardware.d.ts
CHANGED
|
@@ -96,11 +96,39 @@ export declare const SKUS: {
|
|
|
96
96
|
tflops: number;
|
|
97
97
|
memory: number[];
|
|
98
98
|
};
|
|
99
|
+
"RTX 3080": {
|
|
100
|
+
tflops: number;
|
|
101
|
+
memory: number[];
|
|
102
|
+
};
|
|
99
103
|
"RTX 3080 Ti": {
|
|
100
104
|
tflops: number;
|
|
101
105
|
memory: number[];
|
|
102
106
|
};
|
|
103
|
-
"RTX
|
|
107
|
+
"RTX 3070": {
|
|
108
|
+
tflops: number;
|
|
109
|
+
memory: number[];
|
|
110
|
+
};
|
|
111
|
+
"RTX 3070 Ti": {
|
|
112
|
+
tflops: number;
|
|
113
|
+
memory: number[];
|
|
114
|
+
};
|
|
115
|
+
"RTX 3070 Ti Laptop": {
|
|
116
|
+
tflops: number;
|
|
117
|
+
memory: number[];
|
|
118
|
+
};
|
|
119
|
+
"RTX 3060 Ti": {
|
|
120
|
+
tflops: number;
|
|
121
|
+
memory: number[];
|
|
122
|
+
};
|
|
123
|
+
"RTX 3060": {
|
|
124
|
+
tflops: number;
|
|
125
|
+
memory: number[];
|
|
126
|
+
};
|
|
127
|
+
"RTX 2060 Mobile": {
|
|
128
|
+
tflops: number;
|
|
129
|
+
memory: number[];
|
|
130
|
+
};
|
|
131
|
+
"RTX Titan": {
|
|
104
132
|
tflops: number;
|
|
105
133
|
memory: number[];
|
|
106
134
|
};
|
|
@@ -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,
|
|
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;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA2SuD,CAAC;AAEzE,MAAM,MAAM,OAAO,GAAG,MAAM,OAAO,IAAI,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"library-to-tasks.d.ts","sourceRoot":"","sources":["../../src/library-to-tasks.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACzD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAEhD;;;;;;GAMG;AACH,eAAO,MAAM,oBAAoB,EAAE,OAAO,CAAC,MAAM,CAAC,eAAe,EAAE,YAAY,EAAE,CAAC,
|
|
1
|
+
{"version":3,"file":"library-to-tasks.d.ts","sourceRoot":"","sources":["../../src/library-to-tasks.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACzD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAEhD;;;;;;GAMG;AACH,eAAO,MAAM,oBAAoB,EAAE,OAAO,CAAC,MAAM,CAAC,eAAe,EAAE,YAAY,EAAE,CAAC,CAiEjF,CAAC"}
|
|
@@ -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;
|
|
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;AAMF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AA+BF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAUlD,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,EAK9C,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EASlD,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,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,EAwCrD,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,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"}
|
|
@@ -507,6 +507,6 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
|
|
|
507
507
|
};
|
|
508
508
|
};
|
|
509
509
|
export type ModelLibraryKey = keyof typeof MODEL_LIBRARIES_UI_ELEMENTS;
|
|
510
|
-
export declare const ALL_MODEL_LIBRARY_KEYS: ("
|
|
511
|
-
export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("
|
|
510
|
+
export declare const ALL_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "bertopic" | "big_vision" | "diffusers" | "doctr" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gliner" | "grok" | "keras" | "keras-nlp" | "k2" | "mindspore" | "ml-agents" | "mlx" | "mlx-image" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "stable-baselines3" | "stanza" | "tensorflowtts" | "timesfm" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "voicecraft" | "whisperkit")[];
|
|
511
|
+
export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "bertopic" | "big_vision" | "diffusers" | "doctr" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gliner" | "grok" | "keras" | "keras-nlp" | "k2" | "mindspore" | "ml-agents" | "mlx" | "mlx-image" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "stable-baselines3" | "stanza" | "tensorflowtts" | "timesfm" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "voicecraft" | "whisperkit")[];
|
|
512
512
|
//# sourceMappingURL=model-libraries.d.ts.map
|
package/package.json
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@huggingface/tasks",
|
|
3
3
|
"packageManager": "pnpm@8.10.5",
|
|
4
|
-
"version": "0.10.
|
|
4
|
+
"version": "0.10.8",
|
|
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/hardware.ts
CHANGED
|
@@ -31,7 +31,7 @@ export interface HardwareSpec {
|
|
|
31
31
|
memory?: number[];
|
|
32
32
|
}
|
|
33
33
|
|
|
34
|
-
export const DEFAULT_MEMORY_OPTIONS = [8, 16, 24, 32, 40, 48, 64, 80, 96, 128];
|
|
34
|
+
export const DEFAULT_MEMORY_OPTIONS = [8, 16, 24, 32, 40, 48, 64, 80, 96, 128, 256, 512];
|
|
35
35
|
|
|
36
36
|
export const SKUS = {
|
|
37
37
|
GPU: {
|
|
@@ -100,13 +100,41 @@ export const SKUS = {
|
|
|
100
100
|
tflops: 40,
|
|
101
101
|
memory: [24],
|
|
102
102
|
},
|
|
103
|
+
"RTX 3080": {
|
|
104
|
+
tflops: 30.6,
|
|
105
|
+
memory: [12, 10],
|
|
106
|
+
},
|
|
103
107
|
"RTX 3080 Ti": {
|
|
104
108
|
tflops: 34.1,
|
|
105
109
|
memory: [12],
|
|
106
110
|
},
|
|
107
|
-
"RTX
|
|
108
|
-
tflops:
|
|
109
|
-
memory: [
|
|
111
|
+
"RTX 3070": {
|
|
112
|
+
tflops: 20.31,
|
|
113
|
+
memory: [8],
|
|
114
|
+
},
|
|
115
|
+
"RTX 3070 Ti": {
|
|
116
|
+
tflops: 21.75,
|
|
117
|
+
memory: [8],
|
|
118
|
+
},
|
|
119
|
+
"RTX 3070 Ti Laptop": {
|
|
120
|
+
tflops: 16.6,
|
|
121
|
+
memory: [8],
|
|
122
|
+
},
|
|
123
|
+
"RTX 3060 Ti": {
|
|
124
|
+
tflops: 16.2,
|
|
125
|
+
memory: [8],
|
|
126
|
+
},
|
|
127
|
+
"RTX 3060": {
|
|
128
|
+
tflops: 12.74,
|
|
129
|
+
memory: [12, 8],
|
|
130
|
+
},
|
|
131
|
+
"RTX 2060 Mobile": {
|
|
132
|
+
tflops: 9.22,
|
|
133
|
+
memory: [6],
|
|
134
|
+
},
|
|
135
|
+
"RTX Titan": {
|
|
136
|
+
tflops: 32.62,
|
|
137
|
+
memory: [24],
|
|
110
138
|
},
|
|
111
139
|
},
|
|
112
140
|
AMD: {
|
package/src/library-to-tasks.ts
CHANGED
|
@@ -70,6 +70,7 @@ export const LIBRARY_TASK_MAPPING: Partial<Record<ModelLibraryKey, PipelineType[
|
|
|
70
70
|
"visual-question-answering",
|
|
71
71
|
"zero-shot-classification",
|
|
72
72
|
"zero-shot-image-classification",
|
|
73
|
+
"zero-shot-object-detection",
|
|
73
74
|
],
|
|
74
75
|
mindspore: ["image-classification"],
|
|
75
76
|
};
|
|
@@ -1,4 +1,5 @@
|
|
|
1
1
|
import type { ModelData } from "./model-data";
|
|
2
|
+
import { LIBRARY_TASK_MAPPING } from "./library-to-tasks";
|
|
2
3
|
|
|
3
4
|
const TAG_CUSTOM_CODE = "custom_code";
|
|
4
5
|
|
|
@@ -438,7 +439,7 @@ export const transformers = (model: ModelData): string[] => {
|
|
|
438
439
|
].join("\n");
|
|
439
440
|
}
|
|
440
441
|
|
|
441
|
-
if (model.pipeline_tag) {
|
|
442
|
+
if (model.pipeline_tag && LIBRARY_TASK_MAPPING.transformers?.includes(model.pipeline_tag)) {
|
|
442
443
|
const pipelineSnippet = [
|
|
443
444
|
"# Use a pipeline as a high-level helper",
|
|
444
445
|
"from transformers import pipeline",
|