@huggingface/tasks 0.13.13 → 0.13.14

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.
Files changed (23) hide show
  1. package/dist/commonjs/hardware.d.ts +26 -0
  2. package/dist/commonjs/hardware.d.ts.map +1 -1
  3. package/dist/commonjs/hardware.js +27 -1
  4. package/dist/commonjs/model-libraries-snippets.d.ts +1 -1
  5. package/dist/commonjs/model-libraries-snippets.d.ts.map +1 -1
  6. package/dist/commonjs/model-libraries-snippets.js +16 -8
  7. package/dist/commonjs/model-libraries.d.ts +33 -5
  8. package/dist/commonjs/model-libraries.d.ts.map +1 -1
  9. package/dist/commonjs/model-libraries.js +37 -8
  10. package/dist/esm/hardware.d.ts +26 -0
  11. package/dist/esm/hardware.d.ts.map +1 -1
  12. package/dist/esm/hardware.js +27 -1
  13. package/dist/esm/model-libraries-snippets.d.ts +1 -1
  14. package/dist/esm/model-libraries-snippets.d.ts.map +1 -1
  15. package/dist/esm/model-libraries-snippets.js +14 -6
  16. package/dist/esm/model-libraries.d.ts +33 -5
  17. package/dist/esm/model-libraries.d.ts.map +1 -1
  18. package/dist/esm/model-libraries.js +37 -8
  19. package/package.json +1 -1
  20. package/src/hardware.ts +27 -1
  21. package/src/model-libraries-snippets.ts +16 -7
  22. package/src/model-libraries.ts +37 -8
  23. package/src/tasks/image-text-to-text/about.md +36 -24
package/dist/commonjs/hardware.d.ts CHANGED
@@ -68,10 +68,18 @@ export declare const SKUS: {
68
68
  tflops: number;
69
69
  memory: number[];
70
70
  };
71
+ "RTX A6000": {
72
+ tflops: number;
73
+ memory: number[];
74
+ };
71
75
  "RTX A4000": {
72
76
  tflops: number;
73
77
  memory: number[];
74
78
  };
79
+ "RTX A2000": {
80
+ tflops: number;
81
+ memory: number[];
82
+ };
75
83
  A100: {
76
84
  tflops: number;
77
85
  memory: number[];
@@ -299,6 +307,24 @@ export declare const SKUS: {
299
307
  memory: number[];
300
308
  };
301
309
  };
310
+ INTEL: {
311
+ "Arc A750": {
312
+ tflops: number;
313
+ memory: number[];
314
+ };
315
+ "Arc A770": {
316
+ tflops: number;
317
+ memory: number[];
318
+ };
319
+ "Arc B570": {
320
+ tflops: number;
321
+ memory: number[];
322
+ };
323
+ "Arc B580": {
324
+ tflops: number;
325
+ memory: number[];
326
+ };
327
+ };
302
328
  QUALCOMM: {
303
329
  "Snapdragon X Elite X1E-00-1DE": {
304
330
  tflops: number;
package/dist/commonjs/hardware.d.ts.map CHANGED
@@ -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;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAwcuD,CAAC;AAEzE,MAAM,MAAM,OAAO,GAAG,MAAM,OAAO,IAAI,CAAC"}
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;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAkeuD,CAAC;AAEzE,MAAM,MAAM,OAAO,GAAG,MAAM,OAAO,IAAI,CAAC"}
package/dist/commonjs/hardware.js CHANGED
@@ -53,10 +53,18 @@ exports.SKUS = {
53
53
  tflops: 12.0,
54
54
  memory: [16],
55
55
  },
56
+ "RTX A6000": {
57
+ tflops: 38.7,
58
+ memory: [48],
59
+ },
56
60
  "RTX A4000": {
57
61
  tflops: 19.2,
58
62
  memory: [16],
59
63
  },
64
+ "RTX A2000": {
65
+ tflops: 7.987,
66
+ memory: [8, 12],
67
+ },
60
68
  A100: {
61
69
  tflops: 77.97,
62
70
  memory: [80, 40],
@@ -284,6 +292,24 @@ exports.SKUS = {
284
292
  memory: [16],
285
293
  },
286
294
  },
295
+ INTEL: {
296
+ "Arc A750": {
297
+ tflops: 34.41,
298
+ memory: [8],
299
+ },
300
+ "Arc A770": {
301
+ tflops: 39.32,
302
+ memory: [8, 16],
303
+ },
304
+ "Arc B570": {
305
+ tflops: 23.04,
306
+ memory: [10],
307
+ },
308
+ "Arc B580": {
309
+ tflops: 27.34,
310
+ memory: [12],
311
+ },
312
+ },
287
313
  QUALCOMM: {
288
314
  "Snapdragon X Elite X1E-00-1DE": {
289
315
  tflops: 4.6,
@@ -462,7 +488,7 @@ exports.SKUS = {
462
488
  },
463
489
  "Apple M4 Pro": {
464
490
  tflops: 9.2,
465
- memory: [24, 48],
491
+ memory: [24, 48, 64],
466
492
  },
467
493
  "Apple M4 Max": {
468
494
  tflops: 18.4,
package/dist/commonjs/model-libraries-snippets.d.ts CHANGED
@@ -58,7 +58,7 @@ export declare const sana: (model: ModelData) => string[];
58
58
  export declare const vfimamba: (model: ModelData) => string[];
59
59
  export declare const voicecraft: (model: ModelData) => string[];
60
60
  export declare const chattts: () => string[];
61
- export declare const yolov10: (model: ModelData) => string[];
61
+ export declare const ultralytics: (model: ModelData) => string[];
62
62
  export declare const birefnet: (model: ModelData) => string[];
63
63
  export declare const mlx: (model: ModelData) => string[];
64
64
  export declare const mlxim: (model: ModelData) => string[];
package/dist/commonjs/model-libraries-snippets.d.ts.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"file":"model-libraries-snippets.d.ts","sourceRoot":"","sources":["../../src/model-libraries-snippets.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAkBjD,eAAO,MAAM,QAAQ,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;AAaF,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;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAwBlD,CAAC;AAuCF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAUlD,CAAC;AAEF,eAAO,MAAM,YAAY,UAAW,SAAS,KAAG,MAAM,EAwCrD,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,OAAO,UAAW,SAAS,KAAG,MAAM,EAehD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAS9C,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAUlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAalD,CAAC;AAEF,eAAO,MAAM,gBAAgB,UAAW,SAAS,KAAG,MAAM,EA0BzD,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;AASF,eAAO,MAAM,oBAAoB,UAAW,SAAS,KAAG,MAAM,EAoB7D,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,EAiB7C,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,IAAI,UAAW,SAAS,KAAG,MAAM,EAkB7C,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAInD,CAAC;AAEF,eAAO,MAAM,OAAO,QAAO,MAAM,EAYhC,CAAC;AAEF,eAAO,MAAM,OAAO,UAAW,SAAS,KAAG,MAAM,EAOhD,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,SAAS,UAAW,SAAS,KAAG,MAAM,EAIlD,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAQ7C,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAI7C,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AA4BF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAO/C,CAAC;AAGF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAUnD,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAYnC,CAAC;AAEF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAKvD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC"}
1
+ {"version":3,"file":"model-libraries-snippets.d.ts","sourceRoot":"","sources":["../../src/model-libraries-snippets.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAkBjD,eAAO,MAAM,QAAQ,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;AAaF,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;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAwBlD,CAAC;AAuCF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAUlD,CAAC;AAEF,eAAO,MAAM,YAAY,UAAW,SAAS,KAAG,MAAM,EAwCrD,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,OAAO,UAAW,SAAS,KAAG,MAAM,EAehD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAS9C,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAUlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAalD,CAAC;AAEF,eAAO,MAAM,gBAAgB,UAAW,SAAS,KAAG,MAAM,EA0BzD,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;AASF,eAAO,MAAM,oBAAoB,UAAW,SAAS,KAAG,MAAM,EAoB7D,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,EAiB7C,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,IAAI,UAAW,SAAS,KAAG,MAAM,EAkB7C,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAInD,CAAC;AAEF,eAAO,MAAM,OAAO,QAAO,MAAM,EAYhC,CAAC;AAEF,eAAO,MAAM,WAAW,UAAW,SAAS,KAAG,MAAM,EAiBpD,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,SAAS,UAAW,SAAS,KAAG,MAAM,EAIlD,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAQ7C,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAI7C,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AA4BF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAO/C,CAAC;AAEF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAUnD,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAYnC,CAAC;AAEF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAKvD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC"}
package/dist/commonjs/model-libraries-snippets.js CHANGED
@@ -1,7 +1,7 @@
1
1
  "use strict";
2
2
  Object.defineProperty(exports, "__esModule", { value: true });
3
3
  exports.transformersJS = exports.transformers = exports.speechbrain = exports.stanza = exports.span_marker = exports.spacy = exports.setfit = exports.sentenceTransformers = exports.sampleFactory = exports.sam2 = exports.fastai = exports.stable_audio_tools = exports.sklearn = exports.seed_story = exports.saelens = exports.timm = exports.tensorflowtts = exports.relik = exports.pyannote_audio = exports.pyannote_audio_pipeline = exports.paddlenlp = exports.open_clip = exports.mesh_anything = exports.mars5_tts = exports.mamba_ssm = exports.tf_keras = exports.llama_cpp_python = exports.keras_hub = exports.keras_nlp = exports.keras = exports.htrflow = exports.gliner = exports.flair = exports.fairseq = exports.espnet = exports.espnetASR = exports.espnetTTS = exports.edsnlp = exports.cartesia_mlx = exports.cartesia_pytorch = exports.diffusionkit = exports.diffusers = exports.depth_pro = exports.depth_anything_v2 = exports.bm25s = exports.bertopic = exports.audioseal = exports.asteroid = exports.allennlp = exports.adapters = void 0;
4
- exports.hezar = exports.threedtopia_xl = exports.whisperkit = exports.audiocraft = exports.anemoi = exports.pythae = exports.pxia = exports.nemo = exports.model2vec = exports.mlxim = exports.mlx = exports.birefnet = exports.yolov10 = exports.chattts = exports.voicecraft = exports.vfimamba = exports.sana = exports.sentis = exports.mlAgents = exports.stableBaselines3 = exports.fasttext = exports.peft = void 0;
4
+ exports.hezar = exports.threedtopia_xl = exports.whisperkit = exports.audiocraft = exports.anemoi = exports.pythae = exports.pxia = exports.nemo = exports.model2vec = exports.mlxim = exports.mlx = exports.birefnet = exports.ultralytics = exports.chattts = exports.voicecraft = exports.vfimamba = exports.sana = exports.sentis = exports.mlAgents = exports.stableBaselines3 = exports.fasttext = exports.peft = void 0;
5
5
  const library_to_tasks_js_1 = require("./library-to-tasks.js");
6
6
  const inputs_js_1 = require("./snippets/inputs.js");
7
7
  const common_js_1 = require("./snippets/common.js");
@@ -991,15 +991,23 @@ wavs = chat.infer(texts, )
991
991
  torchaudio.save("output1.wav", torch.from_numpy(wavs[0]), 24000)`,
992
992
  ];
993
993
  exports.chattts = chattts;
994
- const yolov10 = (model) => [
995
- `from ultralytics import YOLOv10
994
+ const ultralytics = (model) => {
995
+ // ultralytics models must have a version tag (e.g. `yolov8`)
996
+ const versionTag = model.tags.find((tag) => tag.match(/^yolov\d+$/));
997
+ const className = versionTag ? `YOLOv${versionTag.slice(4)}` : "YOLOvXX";
998
+ const prefix = versionTag
999
+ ? ""
1000
+ : `# Couldn't find a valid YOLO version tag.\n# Replace XX with the correct version.\n`;
1001
+ return [
1002
+ prefix +
1003
+ `from ultralytics import ${className}
996
1004
 
997
- model = YOLOv10.from_pretrained("${model.id}")
1005
+ model = ${className}.from_pretrained("${model.id}")
998
1006
  source = 'http://images.cocodataset.org/val2017/000000039769.jpg'
999
- model.predict(source=source, save=True)
1000
- `,
1001
- ];
1002
- exports.yolov10 = yolov10;
1007
+ model.predict(source=source, save=True)`,
1008
+ ];
1009
+ };
1010
+ exports.ultralytics = ultralytics;
1003
1011
  const birefnet = (model) => [
1004
1012
  `# Option 1: use with transformers
1005
1013
 
package/dist/commonjs/model-libraries.d.ts CHANGED
@@ -162,6 +162,12 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
162
162
  filter: false;
163
163
  countDownloads: string;
164
164
  };
165
+ cosmos: {
166
+ prettyLabel: string;
167
+ repoName: string;
168
+ repoUrl: string;
169
+ countDownloads: string;
170
+ };
165
171
  "cxr-foundation": {
166
172
  prettyLabel: string;
167
173
  repoName: string;
@@ -244,6 +250,13 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
244
250
  filter: false;
245
251
  countDownloads: string;
246
252
  };
253
+ cosyvoice: {
254
+ prettyLabel: string;
255
+ repoName: string;
256
+ repoUrl: string;
257
+ filter: false;
258
+ countDownloads: string;
259
+ };
247
260
  cotracker: {
248
261
  prettyLabel: string;
249
262
  repoName: string;
@@ -750,6 +763,11 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
750
763
  repoUrl: string;
751
764
  snippets: (model: ModelData) => string[];
752
765
  };
766
+ tabpfn: {
767
+ prettyLabel: string;
768
+ repoName: string;
769
+ repoUrl: string;
770
+ };
753
771
  "tic-clip": {
754
772
  prettyLabel: string;
755
773
  repoName: string;
@@ -795,6 +813,15 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
795
813
  repoUrl: string;
796
814
  countDownloads: string;
797
815
  };
816
+ ultralytics: {
817
+ prettyLabel: string;
818
+ repoName: string;
819
+ repoUrl: string;
820
+ docsUrl: string;
821
+ filter: false;
822
+ countDownloads: string;
823
+ snippets: (model: ModelData) => string[];
824
+ };
798
825
  "unity-sentis": {
799
826
  prettyLabel: string;
800
827
  repoName: string;
@@ -824,20 +851,21 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
824
851
  docsUrl: string;
825
852
  snippets: (model: ModelData) => string[];
826
853
  };
827
- yolov10: {
854
+ whisperkit: {
828
855
  prettyLabel: string;
829
856
  repoName: string;
830
857
  repoUrl: string;
831
858
  docsUrl: string;
832
- snippets: (model: ModelData) => string[];
859
+ snippets: () => string[];
860
+ countDownloads: string;
833
861
  };
834
- whisperkit: {
862
+ yolov10: {
835
863
  prettyLabel: string;
836
864
  repoName: string;
837
865
  repoUrl: string;
838
866
  docsUrl: string;
839
- snippets: () => string[];
840
867
  countDownloads: string;
868
+ snippets: (model: ModelData) => string[];
841
869
  };
842
870
  "3dtopia-xl": {
843
871
  prettyLabel: string;
@@ -850,5 +878,5 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
850
878
  };
851
879
  export type ModelLibraryKey = keyof typeof MODEL_LIBRARIES_UI_ELEMENTS;
852
880
  export declare const ALL_MODEL_LIBRARY_KEYS: ModelLibraryKey[];
853
- export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "anemoi" | "asteroid" | "audiocraft" | "audioseal" | "bertopic" | "big_vision" | "birder" | "birefnet" | "bm25s" | "champ" | "chat_tts" | "colpali" | "cxr-foundation" | "deepforest" | "depth-anything-v2" | "depth-pro" | "derm-foundation" | "diffree" | "diffusers" | "diffusionkit" | "doctr" | "cartesia_pytorch" | "cartesia_mlx" | "clipscope" | "cotracker" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gemma.cpp" | "gliner" | "glyph-byt5" | "grok" | "hallo" | "hezar" | "htrflow" | "hunyuan-dit" | "imstoucan" | "keras" | "tf-keras" | "keras-nlp" | "keras-hub" | "k2" | "liveportrait" | "llama-cpp-python" | "mini-omni2" | "mindspore" | "mamba-ssm" | "mars5-tts" | "mesh-anything" | "ml-agents" | "mlx" | "mlx-image" | "mlc-llm" | "model2vec" | "moshi" | "nemo" | "open-oasis" | "open_clip" | "paddlenlp" | "peft" | "pxia" | "pyannote-audio" | "py-feat" | "pythae" | "recurrentgemma" | "relik" | "refiners" | "reverb" | "saelens" | "sam2" | "sample-factory" | "sapiens" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "ssr-speech" | "stable-audio-tools" | "diffusion-single-file" | "seed-story" | "soloaudio" | "stable-baselines3" | "stanza" | "f5-tts" | "genmo" | "tensorflowtts" | "tic-clip" | "timesfm" | "timm" | "transformers" | "transformers.js" | "trellis" | "unity-sentis" | "sana" | "vfi-mamba" | "voicecraft" | "yolov10" | "whisperkit" | "3dtopia-xl")[];
881
+ export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "anemoi" | "asteroid" | "audiocraft" | "audioseal" | "bertopic" | "big_vision" | "birder" | "birefnet" | "bm25s" | "champ" | "chat_tts" | "colpali" | "cosmos" | "cxr-foundation" | "deepforest" | "depth-anything-v2" | "depth-pro" | "derm-foundation" | "diffree" | "diffusers" | "diffusionkit" | "doctr" | "cartesia_pytorch" | "cartesia_mlx" | "clipscope" | "cosyvoice" | "cotracker" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gemma.cpp" | "gliner" | "glyph-byt5" | "grok" | "hallo" | "hezar" | "htrflow" | "hunyuan-dit" | "imstoucan" | "keras" | "tf-keras" | "keras-nlp" | "keras-hub" | "k2" | "liveportrait" | "llama-cpp-python" | "mini-omni2" | "mindspore" | "mamba-ssm" | "mars5-tts" | "mesh-anything" | "ml-agents" | "mlx" | "mlx-image" | "mlc-llm" | "model2vec" | "moshi" | "nemo" | "open-oasis" | "open_clip" | "paddlenlp" | "peft" | "pxia" | "pyannote-audio" | "py-feat" | "pythae" | "recurrentgemma" | "relik" | "refiners" | "reverb" | "saelens" | "sam2" | "sample-factory" | "sapiens" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "ssr-speech" | "stable-audio-tools" | "diffusion-single-file" | "seed-story" | "soloaudio" | "stable-baselines3" | "stanza" | "f5-tts" | "genmo" | "tensorflowtts" | "tabpfn" | "tic-clip" | "timesfm" | "timm" | "transformers" | "transformers.js" | "trellis" | "ultralytics" | "unity-sentis" | "sana" | "vfi-mamba" | "voicecraft" | "whisperkit" | "yolov10" | "3dtopia-xl")[];
854
882
  //# sourceMappingURL=model-libraries.d.ts.map
package/dist/commonjs/model-libraries.d.ts.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"file":"model-libraries.d.ts","sourceRoot":"","sources":["../../src/model-libraries.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AAEzE;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;;;OAIG;IACH,WAAW,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,MAAM,EAAE,CAAC;IAC1C;;;;;OAKG;IACH,cAAc,CAAC,EAAE,kBAAkB,CAAC;IACpC;;;OAGG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;CACjB;AAED;;;;;;;;;;;;;GAaG;AAEH,eAAO,MAAM,2BAA2B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA4xBI,CAAC;AAE7C,MAAM,MAAM,eAAe,GAAG,MAAM,OAAO,2BAA2B,CAAC;AAEvE,eAAO,MAAM,sBAAsB,EAA+C,eAAe,EAAE,CAAC;AAEpG,eAAO,MAAM,8BAA8B,28CAQ1B,CAAC"}
1
+ {"version":3,"file":"model-libraries.d.ts","sourceRoot":"","sources":["../../src/model-libraries.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AAEzE;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;;;OAIG;IACH,WAAW,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,MAAM,EAAE,CAAC;IAC1C;;;;;OAKG;IACH,cAAc,CAAC,EAAE,kBAAkB,CAAC;IACpC;;;OAGG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;CACjB;AAED;;;;;;;;;;;;;GAaG;AAEH,eAAO,MAAM,2BAA2B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAyzBI,CAAC;AAE7C,MAAM,MAAM,eAAe,GAAG,MAAM,OAAO,2BAA2B,CAAC;AAEvE,eAAO,MAAM,sBAAsB,EAA+C,eAAe,EAAE,CAAC;AAEpG,eAAO,MAAM,8BAA8B,+/CAQ1B,CAAC"}
package/dist/commonjs/model-libraries.js CHANGED
@@ -148,6 +148,12 @@ exports.MODEL_LIBRARIES_UI_ELEMENTS = {
148
148
  filter: false,
149
149
  countDownloads: `path:"adapter_config.json"`,
150
150
  },
151
+ cosmos: {
152
+ prettyLabel: "Cosmos",
153
+ repoName: "Cosmos",
154
+ repoUrl: "https://github.com/NVIDIA/Cosmos",
155
+ countDownloads: `path:"config.json" OR path_extension:"pt"`,
156
+ },
151
157
  "cxr-foundation": {
152
158
  prettyLabel: "CXR Foundation",
153
159
  repoName: "cxr-foundation",
@@ -231,6 +237,13 @@ exports.MODEL_LIBRARIES_UI_ELEMENTS = {
231
237
  filter: false,
232
238
  countDownloads: `path_extension:"pt"`,
233
239
  },
240
+ cosyvoice: {
241
+ prettyLabel: "CosyVoice",
242
+ repoName: "CosyVoice",
243
+ repoUrl: "https://github.com/FunAudioLLM/CosyVoice",
244
+ filter: false,
245
+ countDownloads: `path_extension:"onnx" OR path_extension:"pt"`,
246
+ },
234
247
  cotracker: {
235
248
  prettyLabel: "CoTracker",
236
249
  repoName: "CoTracker",
@@ -738,6 +751,11 @@ exports.MODEL_LIBRARIES_UI_ELEMENTS = {
738
751
  repoUrl: "https://github.com/TensorSpeech/TensorFlowTTS",
739
752
  snippets: snippets.tensorflowtts,
740
753
  },
754
+ tabpfn: {
755
+ prettyLabel: "TabPFN",
756
+ repoName: "TabPFN",
757
+ repoUrl: "https://github.com/PriorLabs/TabPFN"
758
+ },
741
759
  "tic-clip": {
742
760
  prettyLabel: "TiC-CLIP",
743
761
  repoName: "TiC-CLIP",
@@ -777,12 +795,21 @@ exports.MODEL_LIBRARIES_UI_ELEMENTS = {
777
795
  snippets: snippets.transformersJS,
778
796
  filter: true,
779
797
  },
780
- "trellis": {
798
+ trellis: {
781
799
  prettyLabel: "Trellis",
782
800
  repoName: "Trellis",
783
801
  repoUrl: "https://github.com/microsoft/TRELLIS",
784
802
  countDownloads: `path_extension:"safetensors"`,
785
803
  },
804
+ ultralytics: {
805
+ prettyLabel: "ultralytics",
806
+ repoName: "ultralytics",
807
+ repoUrl: "https://github.com/ultralytics/ultralytics",
808
+ docsUrl: "https://github.com/ultralytics/ultralytics",
809
+ filter: false,
810
+ countDownloads: `path_extension:"pt"`,
811
+ snippets: snippets.ultralytics,
812
+ },
786
813
  "unity-sentis": {
787
814
  prettyLabel: "unity-sentis",
788
815
  repoName: "unity-sentis",
@@ -812,13 +839,6 @@ exports.MODEL_LIBRARIES_UI_ELEMENTS = {
812
839
  docsUrl: "https://github.com/jasonppy/VoiceCraft",
813
840
  snippets: snippets.voicecraft,
814
841
  },
815
- yolov10: {
816
- prettyLabel: "YOLOv10",
817
- repoName: "yolov10",
818
- repoUrl: "https://github.com/THU-MIG/yolov10",
819
- docsUrl: "https://github.com/THU-MIG/yolov10",
820
- snippets: snippets.yolov10,
821
- },
822
842
  whisperkit: {
823
843
  prettyLabel: "WhisperKit",
824
844
  repoName: "WhisperKit",
@@ -827,6 +847,15 @@ exports.MODEL_LIBRARIES_UI_ELEMENTS = {
827
847
  snippets: snippets.whisperkit,
828
848
  countDownloads: `path_filename:"model" AND path_extension:"mil" AND _exists_:"path_prefix"`,
829
849
  },
850
+ yolov10: {
851
+ // YOLOv10 is a fork of ultraLytics. Code snippets and download count are the same but the repo is different.
852
+ prettyLabel: "YOLOv10",
853
+ repoName: "YOLOv10",
854
+ repoUrl: "https://github.com/THU-MIG/yolov10",
855
+ docsUrl: "https://github.com/THU-MIG/yolov10",
856
+ countDownloads: `path_extension:"pt"`,
857
+ snippets: snippets.ultralytics,
858
+ },
830
859
  "3dtopia-xl": {
831
860
  prettyLabel: "3DTopia-XL",
832
861
  repoName: "3DTopia-XL",
package/dist/esm/hardware.d.ts CHANGED
@@ -68,10 +68,18 @@ export declare const SKUS: {
68
68
  tflops: number;
69
69
  memory: number[];
70
70
  };
71
+ "RTX A6000": {
72
+ tflops: number;
73
+ memory: number[];
74
+ };
71
75
  "RTX A4000": {
72
76
  tflops: number;
73
77
  memory: number[];
74
78
  };
79
+ "RTX A2000": {
80
+ tflops: number;
81
+ memory: number[];
82
+ };
75
83
  A100: {
76
84
  tflops: number;
77
85
  memory: number[];
@@ -299,6 +307,24 @@ export declare const SKUS: {
299
307
  memory: number[];
300
308
  };
301
309
  };
310
+ INTEL: {
311
+ "Arc A750": {
312
+ tflops: number;
313
+ memory: number[];
314
+ };
315
+ "Arc A770": {
316
+ tflops: number;
317
+ memory: number[];
318
+ };
319
+ "Arc B570": {
320
+ tflops: number;
321
+ memory: number[];
322
+ };
323
+ "Arc B580": {
324
+ tflops: number;
325
+ memory: number[];
326
+ };
327
+ };
302
328
  QUALCOMM: {
303
329
  "Snapdragon X Elite X1E-00-1DE": {
304
330
  tflops: number;
package/dist/esm/hardware.d.ts.map CHANGED
@@ -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;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAwcuD,CAAC;AAEzE,MAAM,MAAM,OAAO,GAAG,MAAM,OAAO,IAAI,CAAC"}
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;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAkeuD,CAAC;AAEzE,MAAM,MAAM,OAAO,GAAG,MAAM,OAAO,IAAI,CAAC"}
package/dist/esm/hardware.js CHANGED
@@ -50,10 +50,18 @@ export const SKUS = {
50
50
  tflops: 12.0,
51
51
  memory: [16],
52
52
  },
53
+ "RTX A6000": {
54
+ tflops: 38.7,
55
+ memory: [48],
56
+ },
53
57
  "RTX A4000": {
54
58
  tflops: 19.2,
55
59
  memory: [16],
56
60
  },
61
+ "RTX A2000": {
62
+ tflops: 7.987,
63
+ memory: [8, 12],
64
+ },
57
65
  A100: {
58
66
  tflops: 77.97,
59
67
  memory: [80, 40],
@@ -281,6 +289,24 @@ export const SKUS = {
281
289
  memory: [16],
282
290
  },
283
291
  },
292
+ INTEL: {
293
+ "Arc A750": {
294
+ tflops: 34.41,
295
+ memory: [8],
296
+ },
297
+ "Arc A770": {
298
+ tflops: 39.32,
299
+ memory: [8, 16],
300
+ },
301
+ "Arc B570": {
302
+ tflops: 23.04,
303
+ memory: [10],
304
+ },
305
+ "Arc B580": {
306
+ tflops: 27.34,
307
+ memory: [12],
308
+ },
309
+ },
284
310
  QUALCOMM: {
285
311
  "Snapdragon X Elite X1E-00-1DE": {
286
312
  tflops: 4.6,
@@ -459,7 +485,7 @@ export const SKUS = {
459
485
  },
460
486
  "Apple M4 Pro": {
461
487
  tflops: 9.2,
462
- memory: [24, 48],
488
+ memory: [24, 48, 64],
463
489
  },
464
490
  "Apple M4 Max": {
465
491
  tflops: 18.4,
package/dist/esm/model-libraries-snippets.d.ts CHANGED
@@ -58,7 +58,7 @@ export declare const sana: (model: ModelData) => string[];
58
58
  export declare const vfimamba: (model: ModelData) => string[];
59
59
  export declare const voicecraft: (model: ModelData) => string[];
60
60
  export declare const chattts: () => string[];
61
- export declare const yolov10: (model: ModelData) => string[];
61
+ export declare const ultralytics: (model: ModelData) => string[];
62
62
  export declare const birefnet: (model: ModelData) => string[];
63
63
  export declare const mlx: (model: ModelData) => string[];
64
64
  export declare const mlxim: (model: ModelData) => string[];
package/dist/esm/model-libraries-snippets.d.ts.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"file":"model-libraries-snippets.d.ts","sourceRoot":"","sources":["../../src/model-libraries-snippets.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAkBjD,eAAO,MAAM,QAAQ,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;AAaF,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;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAwBlD,CAAC;AAuCF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAUlD,CAAC;AAEF,eAAO,MAAM,YAAY,UAAW,SAAS,KAAG,MAAM,EAwCrD,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,OAAO,UAAW,SAAS,KAAG,MAAM,EAehD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAS9C,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAUlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAalD,CAAC;AAEF,eAAO,MAAM,gBAAgB,UAAW,SAAS,KAAG,MAAM,EA0BzD,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;AASF,eAAO,MAAM,oBAAoB,UAAW,SAAS,KAAG,MAAM,EAoB7D,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,EAiB7C,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,IAAI,UAAW,SAAS,KAAG,MAAM,EAkB7C,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAInD,CAAC;AAEF,eAAO,MAAM,OAAO,QAAO,MAAM,EAYhC,CAAC;AAEF,eAAO,MAAM,OAAO,UAAW,SAAS,KAAG,MAAM,EAOhD,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,SAAS,UAAW,SAAS,KAAG,MAAM,EAIlD,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAQ7C,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAI7C,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AA4BF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAO/C,CAAC;AAGF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAUnD,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAYnC,CAAC;AAEF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAKvD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC"}
1
+ {"version":3,"file":"model-libraries-snippets.d.ts","sourceRoot":"","sources":["../../src/model-libraries-snippets.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAkBjD,eAAO,MAAM,QAAQ,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;AAaF,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;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAwBlD,CAAC;AAuCF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAUlD,CAAC;AAEF,eAAO,MAAM,YAAY,UAAW,SAAS,KAAG,MAAM,EAwCrD,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,OAAO,UAAW,SAAS,KAAG,MAAM,EAehD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAS9C,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAUlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAalD,CAAC;AAEF,eAAO,MAAM,gBAAgB,UAAW,SAAS,KAAG,MAAM,EA0BzD,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;AASF,eAAO,MAAM,oBAAoB,UAAW,SAAS,KAAG,MAAM,EAoB7D,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,EAiB7C,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,IAAI,UAAW,SAAS,KAAG,MAAM,EAkB7C,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAInD,CAAC;AAEF,eAAO,MAAM,OAAO,QAAO,MAAM,EAYhC,CAAC;AAEF,eAAO,MAAM,WAAW,UAAW,SAAS,KAAG,MAAM,EAiBpD,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,SAAS,UAAW,SAAS,KAAG,MAAM,EAIlD,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAQ7C,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAI7C,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AA4BF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAO/C,CAAC;AAEF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAUnD,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAYnC,CAAC;AAEF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAKvD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC"}
package/dist/esm/model-libraries-snippets.js CHANGED
@@ -928,14 +928,22 @@ wavs = chat.infer(texts, )
928
928
 
929
929
  torchaudio.save("output1.wav", torch.from_numpy(wavs[0]), 24000)`,
930
930
  ];
931
- export const yolov10 = (model) => [
932
- `from ultralytics import YOLOv10
931
+ export const ultralytics = (model) => {
932
+ // ultralytics models must have a version tag (e.g. `yolov8`)
933
+ const versionTag = model.tags.find((tag) => tag.match(/^yolov\d+$/));
934
+ const className = versionTag ? `YOLOv${versionTag.slice(4)}` : "YOLOvXX";
935
+ const prefix = versionTag
936
+ ? ""
937
+ : `# Couldn't find a valid YOLO version tag.\n# Replace XX with the correct version.\n`;
938
+ return [
939
+ prefix +
940
+ `from ultralytics import ${className}
933
941
 
934
- model = YOLOv10.from_pretrained("${model.id}")
942
+ model = ${className}.from_pretrained("${model.id}")
935
943
  source = 'http://images.cocodataset.org/val2017/000000039769.jpg'
936
- model.predict(source=source, save=True)
937
- `,
938
- ];
944
+ model.predict(source=source, save=True)`,
945
+ ];
946
+ };
939
947
  export const birefnet = (model) => [
940
948
  `# Option 1: use with transformers
941
949
 
package/dist/esm/model-libraries.d.ts CHANGED
@@ -162,6 +162,12 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
162
162
  filter: false;
163
163
  countDownloads: string;
164
164
  };
165
+ cosmos: {
166
+ prettyLabel: string;
167
+ repoName: string;
168
+ repoUrl: string;
169
+ countDownloads: string;
170
+ };
165
171
  "cxr-foundation": {
166
172
  prettyLabel: string;
167
173
  repoName: string;
@@ -244,6 +250,13 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
244
250
  filter: false;
245
251
  countDownloads: string;
246
252
  };
253
+ cosyvoice: {
254
+ prettyLabel: string;
255
+ repoName: string;
256
+ repoUrl: string;
257
+ filter: false;
258
+ countDownloads: string;
259
+ };
247
260
  cotracker: {
248
261
  prettyLabel: string;
249
262
  repoName: string;
@@ -750,6 +763,11 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
750
763
  repoUrl: string;
751
764
  snippets: (model: ModelData) => string[];
752
765
  };
766
+ tabpfn: {
767
+ prettyLabel: string;
768
+ repoName: string;
769
+ repoUrl: string;
770
+ };
753
771
  "tic-clip": {
754
772
  prettyLabel: string;
755
773
  repoName: string;
@@ -795,6 +813,15 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
795
813
  repoUrl: string;
796
814
  countDownloads: string;
797
815
  };
816
+ ultralytics: {
817
+ prettyLabel: string;
818
+ repoName: string;
819
+ repoUrl: string;
820
+ docsUrl: string;
821
+ filter: false;
822
+ countDownloads: string;
823
+ snippets: (model: ModelData) => string[];
824
+ };
798
825
  "unity-sentis": {
799
826
  prettyLabel: string;
800
827
  repoName: string;
@@ -824,20 +851,21 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
824
851
  docsUrl: string;
825
852
  snippets: (model: ModelData) => string[];
826
853
  };
827
- yolov10: {
854
+ whisperkit: {
828
855
  prettyLabel: string;
829
856
  repoName: string;
830
857
  repoUrl: string;
831
858
  docsUrl: string;
832
- snippets: (model: ModelData) => string[];
859
+ snippets: () => string[];
860
+ countDownloads: string;
833
861
  };
834
- whisperkit: {
862
+ yolov10: {
835
863
  prettyLabel: string;
836
864
  repoName: string;
837
865
  repoUrl: string;
838
866
  docsUrl: string;
839
- snippets: () => string[];
840
867
  countDownloads: string;
868
+ snippets: (model: ModelData) => string[];
841
869
  };
842
870
  "3dtopia-xl": {
843
871
  prettyLabel: string;
@@ -850,5 +878,5 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
850
878
  };
851
879
  export type ModelLibraryKey = keyof typeof MODEL_LIBRARIES_UI_ELEMENTS;
852
880
  export declare const ALL_MODEL_LIBRARY_KEYS: ModelLibraryKey[];
853
- export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "anemoi" | "asteroid" | "audiocraft" | "audioseal" | "bertopic" | "big_vision" | "birder" | "birefnet" | "bm25s" | "champ" | "chat_tts" | "colpali" | "cxr-foundation" | "deepforest" | "depth-anything-v2" | "depth-pro" | "derm-foundation" | "diffree" | "diffusers" | "diffusionkit" | "doctr" | "cartesia_pytorch" | "cartesia_mlx" | "clipscope" | "cotracker" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gemma.cpp" | "gliner" | "glyph-byt5" | "grok" | "hallo" | "hezar" | "htrflow" | "hunyuan-dit" | "imstoucan" | "keras" | "tf-keras" | "keras-nlp" | "keras-hub" | "k2" | "liveportrait" | "llama-cpp-python" | "mini-omni2" | "mindspore" | "mamba-ssm" | "mars5-tts" | "mesh-anything" | "ml-agents" | "mlx" | "mlx-image" | "mlc-llm" | "model2vec" | "moshi" | "nemo" | "open-oasis" | "open_clip" | "paddlenlp" | "peft" | "pxia" | "pyannote-audio" | "py-feat" | "pythae" | "recurrentgemma" | "relik" | "refiners" | "reverb" | "saelens" | "sam2" | "sample-factory" | "sapiens" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "ssr-speech" | "stable-audio-tools" | "diffusion-single-file" | "seed-story" | "soloaudio" | "stable-baselines3" | "stanza" | "f5-tts" | "genmo" | "tensorflowtts" | "tic-clip" | "timesfm" | "timm" | "transformers" | "transformers.js" | "trellis" | "unity-sentis" | "sana" | "vfi-mamba" | "voicecraft" | "yolov10" | "whisperkit" | "3dtopia-xl")[];
881
+ export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "anemoi" | "asteroid" | "audiocraft" | "audioseal" | "bertopic" | "big_vision" | "birder" | "birefnet" | "bm25s" | "champ" | "chat_tts" | "colpali" | "cosmos" | "cxr-foundation" | "deepforest" | "depth-anything-v2" | "depth-pro" | "derm-foundation" | "diffree" | "diffusers" | "diffusionkit" | "doctr" | "cartesia_pytorch" | "cartesia_mlx" | "clipscope" | "cosyvoice" | "cotracker" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gemma.cpp" | "gliner" | "glyph-byt5" | "grok" | "hallo" | "hezar" | "htrflow" | "hunyuan-dit" | "imstoucan" | "keras" | "tf-keras" | "keras-nlp" | "keras-hub" | "k2" | "liveportrait" | "llama-cpp-python" | "mini-omni2" | "mindspore" | "mamba-ssm" | "mars5-tts" | "mesh-anything" | "ml-agents" | "mlx" | "mlx-image" | "mlc-llm" | "model2vec" | "moshi" | "nemo" | "open-oasis" | "open_clip" | "paddlenlp" | "peft" | "pxia" | "pyannote-audio" | "py-feat" | "pythae" | "recurrentgemma" | "relik" | "refiners" | "reverb" | "saelens" | "sam2" | "sample-factory" | "sapiens" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "ssr-speech" | "stable-audio-tools" | "diffusion-single-file" | "seed-story" | "soloaudio" | "stable-baselines3" | "stanza" | "f5-tts" | "genmo" | "tensorflowtts" | "tabpfn" | "tic-clip" | "timesfm" | "timm" | "transformers" | "transformers.js" | "trellis" | "ultralytics" | "unity-sentis" | "sana" | "vfi-mamba" | "voicecraft" | "whisperkit" | "yolov10" | "3dtopia-xl")[];
854
882
  //# sourceMappingURL=model-libraries.d.ts.map
package/dist/esm/model-libraries.d.ts.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"file":"model-libraries.d.ts","sourceRoot":"","sources":["../../src/model-libraries.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AAEzE;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;;;OAIG;IACH,WAAW,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,MAAM,EAAE,CAAC;IAC1C;;;;;OAKG;IACH,cAAc,CAAC,EAAE,kBAAkB,CAAC;IACpC;;;OAGG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;CACjB;AAED;;;;;;;;;;;;;GAaG;AAEH,eAAO,MAAM,2BAA2B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA4xBI,CAAC;AAE7C,MAAM,MAAM,eAAe,GAAG,MAAM,OAAO,2BAA2B,CAAC;AAEvE,eAAO,MAAM,sBAAsB,EAA+C,eAAe,EAAE,CAAC;AAEpG,eAAO,MAAM,8BAA8B,28CAQ1B,CAAC"}
1
+ {"version":3,"file":"model-libraries.d.ts","sourceRoot":"","sources":["../../src/model-libraries.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AAEzE;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;;;OAIG;IACH,WAAW,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,MAAM,EAAE,CAAC;IAC1C;;;;;OAKG;IACH,cAAc,CAAC,EAAE,kBAAkB,CAAC;IACpC;;;OAGG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;CACjB;AAED;;;;;;;;;;;;;GAaG;AAEH,eAAO,MAAM,2BAA2B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAyzBI,CAAC;AAE7C,MAAM,MAAM,eAAe,GAAG,MAAM,OAAO,2BAA2B,CAAC;AAEvE,eAAO,MAAM,sBAAsB,EAA+C,eAAe,EAAE,CAAC;AAEpG,eAAO,MAAM,8BAA8B,+/CAQ1B,CAAC"}
package/dist/esm/model-libraries.js CHANGED
@@ -122,6 +122,12 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
122
122
  filter: false,
123
123
  countDownloads: `path:"adapter_config.json"`,
124
124
  },
125
+ cosmos: {
126
+ prettyLabel: "Cosmos",
127
+ repoName: "Cosmos",
128
+ repoUrl: "https://github.com/NVIDIA/Cosmos",
129
+ countDownloads: `path:"config.json" OR path_extension:"pt"`,
130
+ },
125
131
  "cxr-foundation": {
126
132
  prettyLabel: "CXR Foundation",
127
133
  repoName: "cxr-foundation",
@@ -205,6 +211,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
205
211
  filter: false,
206
212
  countDownloads: `path_extension:"pt"`,
207
213
  },
214
+ cosyvoice: {
215
+ prettyLabel: "CosyVoice",
216
+ repoName: "CosyVoice",
217
+ repoUrl: "https://github.com/FunAudioLLM/CosyVoice",
218
+ filter: false,
219
+ countDownloads: `path_extension:"onnx" OR path_extension:"pt"`,
220
+ },
208
221
  cotracker: {
209
222
  prettyLabel: "CoTracker",
210
223
  repoName: "CoTracker",
@@ -712,6 +725,11 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
712
725
  repoUrl: "https://github.com/TensorSpeech/TensorFlowTTS",
713
726
  snippets: snippets.tensorflowtts,
714
727
  },
728
+ tabpfn: {
729
+ prettyLabel: "TabPFN",
730
+ repoName: "TabPFN",
731
+ repoUrl: "https://github.com/PriorLabs/TabPFN"
732
+ },
715
733
  "tic-clip": {
716
734
  prettyLabel: "TiC-CLIP",
717
735
  repoName: "TiC-CLIP",
@@ -751,12 +769,21 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
751
769
  snippets: snippets.transformersJS,
752
770
  filter: true,
753
771
  },
754
- "trellis": {
772
+ trellis: {
755
773
  prettyLabel: "Trellis",
756
774
  repoName: "Trellis",
757
775
  repoUrl: "https://github.com/microsoft/TRELLIS",
758
776
  countDownloads: `path_extension:"safetensors"`,
759
777
  },
778
+ ultralytics: {
779
+ prettyLabel: "ultralytics",
780
+ repoName: "ultralytics",
781
+ repoUrl: "https://github.com/ultralytics/ultralytics",
782
+ docsUrl: "https://github.com/ultralytics/ultralytics",
783
+ filter: false,
784
+ countDownloads: `path_extension:"pt"`,
785
+ snippets: snippets.ultralytics,
786
+ },
760
787
  "unity-sentis": {
761
788
  prettyLabel: "unity-sentis",
762
789
  repoName: "unity-sentis",
@@ -786,13 +813,6 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
786
813
  docsUrl: "https://github.com/jasonppy/VoiceCraft",
787
814
  snippets: snippets.voicecraft,
788
815
  },
789
- yolov10: {
790
- prettyLabel: "YOLOv10",
791
- repoName: "yolov10",
792
- repoUrl: "https://github.com/THU-MIG/yolov10",
793
- docsUrl: "https://github.com/THU-MIG/yolov10",
794
- snippets: snippets.yolov10,
795
- },
796
816
  whisperkit: {
797
817
  prettyLabel: "WhisperKit",
798
818
  repoName: "WhisperKit",
@@ -801,6 +821,15 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
801
821
  snippets: snippets.whisperkit,
802
822
  countDownloads: `path_filename:"model" AND path_extension:"mil" AND _exists_:"path_prefix"`,
803
823
  },
824
+ yolov10: {
825
+ // YOLOv10 is a fork of ultraLytics. Code snippets and download count are the same but the repo is different.
826
+ prettyLabel: "YOLOv10",
827
+ repoName: "YOLOv10",
828
+ repoUrl: "https://github.com/THU-MIG/yolov10",
829
+ docsUrl: "https://github.com/THU-MIG/yolov10",
830
+ countDownloads: `path_extension:"pt"`,
831
+ snippets: snippets.ultralytics,
832
+ },
804
833
  "3dtopia-xl": {
805
834
  prettyLabel: "3DTopia-XL",
806
835
  repoName: "3DTopia-XL",
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.13.13",
4
+ "version": "0.13.14",
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
@@ -72,10 +72,18 @@ export const SKUS = {
72
72
  tflops: 12.0,
73
73
  memory: [16],
74
74
  },
75
+ "RTX A6000": {
76
+ tflops: 38.7,
77
+ memory: [48],
78
+ },
75
79
  "RTX A4000": {
76
80
  tflops: 19.2,
77
81
  memory: [16],
78
82
  },
83
+ "RTX A2000": {
84
+ tflops: 7.987,
85
+ memory: [8, 12],
86
+ },
79
87
  A100: {
80
88
  tflops: 77.97,
81
89
  memory: [80, 40],
@@ -303,6 +311,24 @@ export const SKUS = {
303
311
  memory: [16],
304
312
  },
305
313
  },
314
+ INTEL: {
315
+ "Arc A750": {
316
+ tflops: 34.41,
317
+ memory: [8],
318
+ },
319
+ "Arc A770": {
320
+ tflops: 39.32,
321
+ memory: [8, 16],
322
+ },
323
+ "Arc B570": {
324
+ tflops: 23.04,
325
+ memory: [10],
326
+ },
327
+ "Arc B580": {
328
+ tflops: 27.34,
329
+ memory: [12],
330
+ },
331
+ },
306
332
  QUALCOMM: {
307
333
  "Snapdragon X Elite X1E-00-1DE": {
308
334
  tflops: 4.6,
@@ -481,7 +507,7 @@ export const SKUS = {
481
507
  },
482
508
  "Apple M4 Pro": {
483
509
  tflops: 9.2,
484
- memory: [24, 48],
510
+ memory: [24, 48, 64],
485
511
  },
486
512
  "Apple M4 Max": {
487
513
  tflops: 18.4,
package/src/model-libraries-snippets.ts CHANGED
@@ -1028,14 +1028,24 @@ wavs = chat.infer(texts, )
1028
1028
  torchaudio.save("output1.wav", torch.from_numpy(wavs[0]), 24000)`,
1029
1029
  ];
1030
1030
 
1031
- export const yolov10 = (model: ModelData): string[] => [
1032
- `from ultralytics import YOLOv10
1031
+ export const ultralytics = (model: ModelData): string[] => {
1032
+ // ultralytics models must have a version tag (e.g. `yolov8`)
1033
+ const versionTag = model.tags.find((tag) => tag.match(/^yolov\d+$/));
1033
1034
 
1034
- model = YOLOv10.from_pretrained("${model.id}")
1035
+ const className = versionTag ? `YOLOv${versionTag.slice(4)}` : "YOLOvXX";
1036
+ const prefix = versionTag
1037
+ ? ""
1038
+ : `# Couldn't find a valid YOLO version tag.\n# Replace XX with the correct version.\n`;
1039
+
1040
+ return [
1041
+ prefix +
1042
+ `from ultralytics import ${className}
1043
+
1044
+ model = ${className}.from_pretrained("${model.id}")
1035
1045
  source = 'http://images.cocodataset.org/val2017/000000039769.jpg'
1036
- model.predict(source=source, save=True)
1037
- `,
1038
- ];
1046
+ model.predict(source=source, save=True)`,
1047
+ ];
1048
+ };
1039
1049
 
1040
1050
  export const birefnet = (model: ModelData): string[] => [
1041
1051
  `# Option 1: use with transformers
@@ -1127,7 +1137,6 @@ config = Configuration(checkpoint = {"huggingface":{"repo_id":"${model.id}"}})
1127
1137
  runner = DefaultRunner(config)`,
1128
1138
  ];
1129
1139
 
1130
-
1131
1140
  export const audiocraft = (model: ModelData): string[] => {
1132
1141
  if (model.tags.includes("musicgen")) {
1133
1142
  return musicgen(model);
package/src/model-libraries.ts CHANGED
@@ -166,6 +166,12 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
166
166
  filter: false,
167
167
  countDownloads: `path:"adapter_config.json"`,
168
168
  },
169
+ cosmos: {
170
+ prettyLabel: "Cosmos",
171
+ repoName: "Cosmos",
172
+ repoUrl: "https://github.com/NVIDIA/Cosmos",
173
+ countDownloads: `path:"config.json" OR path_extension:"pt"`,
174
+ },
169
175
  "cxr-foundation": {
170
176
  prettyLabel: "CXR Foundation",
171
177
  repoName: "cxr-foundation",
@@ -249,6 +255,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
249
255
  filter: false,
250
256
  countDownloads: `path_extension:"pt"`,
251
257
  },
258
+ cosyvoice: {
259
+ prettyLabel: "CosyVoice",
260
+ repoName: "CosyVoice",
261
+ repoUrl: "https://github.com/FunAudioLLM/CosyVoice",
262
+ filter: false,
263
+ countDownloads: `path_extension:"onnx" OR path_extension:"pt"`,
264
+ },
252
265
  cotracker: {
253
266
  prettyLabel: "CoTracker",
254
267
  repoName: "CoTracker",
@@ -756,6 +769,11 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
756
769
  repoUrl: "https://github.com/TensorSpeech/TensorFlowTTS",
757
770
  snippets: snippets.tensorflowtts,
758
771
  },
772
+ tabpfn: {
773
+ prettyLabel: "TabPFN",
774
+ repoName: "TabPFN",
775
+ repoUrl: "https://github.com/PriorLabs/TabPFN"
776
+ },
759
777
  "tic-clip": {
760
778
  prettyLabel: "TiC-CLIP",
761
779
  repoName: "TiC-CLIP",
@@ -795,12 +813,21 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
795
813
  snippets: snippets.transformersJS,
796
814
  filter: true,
797
815
  },
798
- "trellis": {
816
+ trellis: {
799
817
  prettyLabel: "Trellis",
800
818
  repoName: "Trellis",
801
819
  repoUrl: "https://github.com/microsoft/TRELLIS",
802
820
  countDownloads: `path_extension:"safetensors"`,
803
821
  },
822
+ ultralytics: {
823
+ prettyLabel: "ultralytics",
824
+ repoName: "ultralytics",
825
+ repoUrl: "https://github.com/ultralytics/ultralytics",
826
+ docsUrl: "https://github.com/ultralytics/ultralytics",
827
+ filter: false,
828
+ countDownloads: `path_extension:"pt"`,
829
+ snippets: snippets.ultralytics,
830
+ },
804
831
  "unity-sentis": {
805
832
  prettyLabel: "unity-sentis",
806
833
  repoName: "unity-sentis",
@@ -830,13 +857,6 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
830
857
  docsUrl: "https://github.com/jasonppy/VoiceCraft",
831
858
  snippets: snippets.voicecraft,
832
859
  },
833
- yolov10: {
834
- prettyLabel: "YOLOv10",
835
- repoName: "yolov10",
836
- repoUrl: "https://github.com/THU-MIG/yolov10",
837
- docsUrl: "https://github.com/THU-MIG/yolov10",
838
- snippets: snippets.yolov10,
839
- },
840
860
  whisperkit: {
841
861
  prettyLabel: "WhisperKit",
842
862
  repoName: "WhisperKit",
@@ -845,6 +865,15 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
845
865
  snippets: snippets.whisperkit,
846
866
  countDownloads: `path_filename:"model" AND path_extension:"mil" AND _exists_:"path_prefix"`,
847
867
  },
868
+ yolov10: {
869
+ // YOLOv10 is a fork of ultraLytics. Code snippets and download count are the same but the repo is different.
870
+ prettyLabel: "YOLOv10",
871
+ repoName: "YOLOv10",
872
+ repoUrl: "https://github.com/THU-MIG/yolov10",
873
+ docsUrl: "https://github.com/THU-MIG/yolov10",
874
+ countDownloads: `path_extension:"pt"`,
875
+ snippets: snippets.ultralytics,
876
+ },
848
877
  "3dtopia-xl": {
849
878
  prettyLabel: "3DTopia-XL",
850
879
  repoName: "3DTopia-XL",
package/src/tasks/image-text-to-text/about.md CHANGED
@@ -32,39 +32,51 @@ Vision language models can recognize images through descriptions. When given det
32
32
 
33
33
  ## Inference
34
34
 
35
- You can use the Transformers library to interact with vision-language models. You can load the model like below.
35
+ You can use the Transformers library to interact with [vision-language models](https://huggingface.co/models?pipeline_tag=image-text-to-text&transformers). Specifically, `pipeline` makes it easy to infer models.
36
+
37
+ Initialize the pipeline first.
38
+
39
+ ```python
40
+ from transformers import pipeline
41
+
42
+ pipe = pipeline("image-text-to-text", model="llava-hf/llava-interleave-qwen-0.5b-hf")
43
+ ```
44
+
45
+ The model's built-in chat template will be used to format the conversational input. We can pass the image as an URL in the `content` part of the user message:
36
46
 
37
47
  ```python
38
- from transformers import LlavaNextProcessor, LlavaNextForConditionalGeneration
39
- import torch
40
-
41
- device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
42
- processor = LlavaNextProcessor.from_pretrained("llava-hf/llava-v1.6-mistral-7b-hf")
43
- model = LlavaNextForConditionalGeneration.from_pretrained(
44
- "llava-hf/llava-v1.6-mistral-7b-hf",
45
- torch_dtype=torch.float16
46
- )
47
- model.to(device)
48
+ messages = [
49
+ {
50
+ "role": "user",
51
+ "content": [
52
+ {
53
+ "type": "image",
54
+ "image": "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/bee.jpg",
55
+ },
56
+ {"type": "text", "text": "Describe this image."},
57
+ ],
58
+ }
59
+ ]
60
+
48
61
  ```
49
62
 
50
- We can infer by passing image and text dialogues.
63
+ We can now directly pass in the messages to the pipeline to infer. The `return_full_text` flag is used to return the full prompt in the response, including the user input. Here we pass `False` to only return the generated text.
51
64
 
52
65
  ```python
53
- from PIL import Image
54
- import requests
66
+ outputs = pipe(text=messages, max_new_tokens=60, return_full_text=False)
55
67
 
56
- # image of a radar chart
57
- url = "https://github.com/haotian-liu/LLaVA/blob/1a91fc274d7c35a9b50b3cb29c4247ae5837ce39/images/llava_v1_5_radar.jpg?raw=true"
58
- image = Image.open(requests.get(url, stream=True).raw)
59
- prompt = "[INST] <image>\nWhat is shown in this image? [/INST]"
68
+ outputs[0]["generated_text"]
69
+ # The image captures a moment of tranquility in nature. At the center of the frame, a pink flower with a yellow center is in full bloom. The flower is surrounded by a cluster of red flowers, their vibrant color contrasting with the pink of the flower. \n\nA black and yellow bee is per
70
+ ```
60
71
 
61
- inputs = processor(prompt, image, return_tensors="pt").to(device)
62
- output = model.generate(**inputs, max_new_tokens=100)
72
+ You can also use the Inference API to test image-text-to-text models. You need to use a [Hugging Face token](https://huggingface.co/settings/tokens) for authentication.
63
73
 
64
- print(processor.decode(output[0], skip_special_tokens=True))
65
- # The image appears to be a radar chart, which is a type of multivariate chart that displays values for multiple variables represented on axes
66
- # starting from the same point. This particular radar chart is showing the performance of different models or systems across various metrics.
67
- # The axes represent different metrics or benchmarks, such as MM-Vet, MM-Vet, MM-Vet, MM-Vet, MM-Vet, MM-V
74
+ ```bash
75
+ curl https://api-inference.huggingface.co/models/meta-llama/Llama-3.2-11B-Vision-Instruct \
76
+ -X POST \
77
+ -d '{"messages": [{"role": "user","content": [{"type": "image"}, {"type": "text", "text": "Can you describe the image?"}]}]}' \
78
+ -H "Content-Type: application/json" \
79
+ -H "Authorization: Bearer hf_***"
68
80
  ```
69
81
 
70
82
  ## Useful Resources