@huggingface/tasks 0.13.13 → 0.13.15

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 +30 -1
  2. package/dist/commonjs/hardware.d.ts.map +1 -1
  3. package/dist/commonjs/hardware.js +30 -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 +39 -5
  8. package/dist/commonjs/model-libraries.d.ts.map +1 -1
  9. package/dist/commonjs/model-libraries.js +43 -8
  10. package/dist/esm/hardware.d.ts +30 -1
  11. package/dist/esm/hardware.d.ts.map +1 -1
  12. package/dist/esm/hardware.js +30 -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 +39 -5
  17. package/dist/esm/model-libraries.d.ts.map +1 -1
  18. package/dist/esm/model-libraries.js +43 -8
  19. package/package.json +1 -1
  20. package/src/hardware.ts +31 -2
  21. package/src/model-libraries-snippets.ts +16 -7
  22. package/src/model-libraries.ts +43 -8
  23. package/src/tasks/image-text-to-text/about.md +36 -24
package/dist/commonjs/hardware.d.ts CHANGED
@@ -12,7 +12,7 @@ export declare const TFLOPS_THRESHOLD_WHITE_HOUSE_CLUSTER: number;
12
12
  export declare const TFLOPS_THRESHOLD_EU_AI_ACT_MODEL_TRAINING_TOTAL: number;
13
13
  export interface HardwareSpec {
14
14
  /**
15
- * Approximate value, in FP16 whenever possible.
15
+ * Approximate value, in FP16 whenever possible for GPUs and FP32 for CPUs.
16
16
  * This is only approximate/theoretical and shouldn't be taken too seriously.
17
17
  * Currently the CPU values are from cpu-monkey.com
18
18
  * while the GPU values are from techpowerup.com
@@ -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;
@@ -390,6 +416,9 @@ export declare const SKUS: {
390
416
  "EPYC 1st Generation (Naples)": {
391
417
  tflops: number;
392
418
  };
419
+ "Ryzen Zen 4 7000 (Threadripper)": {
420
+ tflops: number;
421
+ };
393
422
  "Ryzen Zen4 7000 (Ryzen 9)": {
394
423
  tflops: number;
395
424
  };
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;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAqeuD,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,
@@ -375,6 +401,9 @@ exports.SKUS = {
375
401
  "EPYC 1st Generation (Naples)": {
376
402
  tflops: 0.6,
377
403
  },
404
+ "Ryzen Zen 4 7000 (Threadripper)": {
405
+ tflops: 10.0,
406
+ },
378
407
  "Ryzen Zen4 7000 (Ryzen 9)": {
379
408
  tflops: 0.56,
380
409
  },
@@ -462,7 +491,7 @@ exports.SKUS = {
462
491
  },
463
492
  "Apple M4 Pro": {
464
493
  tflops: 9.2,
465
- memory: [24, 48],
494
+ memory: [24, 48, 64],
466
495
  },
467
496
  "Apple M4 Max": {
468
497
  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;
@@ -452,6 +465,12 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
452
465
  countDownloads: string;
453
466
  snippets: () => string[];
454
467
  };
468
+ mitie: {
469
+ prettyLabel: string;
470
+ repoName: string;
471
+ repoUrl: string;
472
+ countDownloads: string;
473
+ };
455
474
  "ml-agents": {
456
475
  prettyLabel: string;
457
476
  repoName: string;
@@ -750,6 +769,11 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
750
769
  repoUrl: string;
751
770
  snippets: (model: ModelData) => string[];
752
771
  };
772
+ tabpfn: {
773
+ prettyLabel: string;
774
+ repoName: string;
775
+ repoUrl: string;
776
+ };
753
777
  "tic-clip": {
754
778
  prettyLabel: string;
755
779
  repoName: string;
@@ -795,6 +819,15 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
795
819
  repoUrl: string;
796
820
  countDownloads: string;
797
821
  };
822
+ ultralytics: {
823
+ prettyLabel: string;
824
+ repoName: string;
825
+ repoUrl: string;
826
+ docsUrl: string;
827
+ filter: false;
828
+ countDownloads: string;
829
+ snippets: (model: ModelData) => string[];
830
+ };
798
831
  "unity-sentis": {
799
832
  prettyLabel: string;
800
833
  repoName: string;
@@ -824,20 +857,21 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
824
857
  docsUrl: string;
825
858
  snippets: (model: ModelData) => string[];
826
859
  };
827
- yolov10: {
860
+ whisperkit: {
828
861
  prettyLabel: string;
829
862
  repoName: string;
830
863
  repoUrl: string;
831
864
  docsUrl: string;
832
- snippets: (model: ModelData) => string[];
865
+ snippets: () => string[];
866
+ countDownloads: string;
833
867
  };
834
- whisperkit: {
868
+ yolov10: {
835
869
  prettyLabel: string;
836
870
  repoName: string;
837
871
  repoUrl: string;
838
872
  docsUrl: string;
839
- snippets: () => string[];
840
873
  countDownloads: string;
874
+ snippets: (model: ModelData) => string[];
841
875
  };
842
876
  "3dtopia-xl": {
843
877
  prettyLabel: string;
@@ -850,5 +884,5 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
850
884
  };
851
885
  export type ModelLibraryKey = keyof typeof MODEL_LIBRARIES_UI_ELEMENTS;
852
886
  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")[];
887
+ 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" | "mitie" | "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
888
  //# 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;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA+zBI,CAAC;AAE7C,MAAM,MAAM,eAAe,GAAG,MAAM,OAAO,2BAA2B,CAAC;AAEvE,eAAO,MAAM,sBAAsB,EAA+C,eAAe,EAAE,CAAC;AAEpG,eAAO,MAAM,8BAA8B,ygDAQ1B,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",
@@ -440,6 +453,12 @@ exports.MODEL_LIBRARIES_UI_ELEMENTS = {
440
453
  countDownloads: `path:"MeshAnything_350m.pth"`,
441
454
  snippets: snippets.mesh_anything,
442
455
  },
456
+ mitie: {
457
+ prettyLabel: "MITIE",
458
+ repoName: "MITIE",
459
+ repoUrl: "https://github.com/mit-nlp/MITIE",
460
+ countDownloads: `path_filename:"total_word_feature_extractor"`,
461
+ },
443
462
  "ml-agents": {
444
463
  prettyLabel: "ml-agents",
445
464
  repoName: "ml-agents",
@@ -738,6 +757,11 @@ exports.MODEL_LIBRARIES_UI_ELEMENTS = {
738
757
  repoUrl: "https://github.com/TensorSpeech/TensorFlowTTS",
739
758
  snippets: snippets.tensorflowtts,
740
759
  },
760
+ tabpfn: {
761
+ prettyLabel: "TabPFN",
762
+ repoName: "TabPFN",
763
+ repoUrl: "https://github.com/PriorLabs/TabPFN",
764
+ },
741
765
  "tic-clip": {
742
766
  prettyLabel: "TiC-CLIP",
743
767
  repoName: "TiC-CLIP",
@@ -777,12 +801,21 @@ exports.MODEL_LIBRARIES_UI_ELEMENTS = {
777
801
  snippets: snippets.transformersJS,
778
802
  filter: true,
779
803
  },
780
- "trellis": {
804
+ trellis: {
781
805
  prettyLabel: "Trellis",
782
806
  repoName: "Trellis",
783
807
  repoUrl: "https://github.com/microsoft/TRELLIS",
784
808
  countDownloads: `path_extension:"safetensors"`,
785
809
  },
810
+ ultralytics: {
811
+ prettyLabel: "ultralytics",
812
+ repoName: "ultralytics",
813
+ repoUrl: "https://github.com/ultralytics/ultralytics",
814
+ docsUrl: "https://github.com/ultralytics/ultralytics",
815
+ filter: false,
816
+ countDownloads: `path_extension:"pt"`,
817
+ snippets: snippets.ultralytics,
818
+ },
786
819
  "unity-sentis": {
787
820
  prettyLabel: "unity-sentis",
788
821
  repoName: "unity-sentis",
@@ -812,13 +845,6 @@ exports.MODEL_LIBRARIES_UI_ELEMENTS = {
812
845
  docsUrl: "https://github.com/jasonppy/VoiceCraft",
813
846
  snippets: snippets.voicecraft,
814
847
  },
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
848
  whisperkit: {
823
849
  prettyLabel: "WhisperKit",
824
850
  repoName: "WhisperKit",
@@ -827,6 +853,15 @@ exports.MODEL_LIBRARIES_UI_ELEMENTS = {
827
853
  snippets: snippets.whisperkit,
828
854
  countDownloads: `path_filename:"model" AND path_extension:"mil" AND _exists_:"path_prefix"`,
829
855
  },
856
+ yolov10: {
857
+ // YOLOv10 is a fork of ultraLytics. Code snippets and download count are the same but the repo is different.
858
+ prettyLabel: "YOLOv10",
859
+ repoName: "YOLOv10",
860
+ repoUrl: "https://github.com/THU-MIG/yolov10",
861
+ docsUrl: "https://github.com/THU-MIG/yolov10",
862
+ countDownloads: `path_extension:"pt"`,
863
+ snippets: snippets.ultralytics,
864
+ },
830
865
  "3dtopia-xl": {
831
866
  prettyLabel: "3DTopia-XL",
832
867
  repoName: "3DTopia-XL",
package/dist/esm/hardware.d.ts CHANGED
@@ -12,7 +12,7 @@ export declare const TFLOPS_THRESHOLD_WHITE_HOUSE_CLUSTER: number;
12
12
  export declare const TFLOPS_THRESHOLD_EU_AI_ACT_MODEL_TRAINING_TOTAL: number;
13
13
  export interface HardwareSpec {
14
14
  /**
15
- * Approximate value, in FP16 whenever possible.
15
+ * Approximate value, in FP16 whenever possible for GPUs and FP32 for CPUs.
16
16
  * This is only approximate/theoretical and shouldn't be taken too seriously.
17
17
  * Currently the CPU values are from cpu-monkey.com
18
18
  * while the GPU values are from techpowerup.com
@@ -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;
@@ -390,6 +416,9 @@ export declare const SKUS: {
390
416
  "EPYC 1st Generation (Naples)": {
391
417
  tflops: number;
392
418
  };
419
+ "Ryzen Zen 4 7000 (Threadripper)": {
420
+ tflops: number;
421
+ };
393
422
  "Ryzen Zen4 7000 (Ryzen 9)": {
394
423
  tflops: number;
395
424
  };
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;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAqeuD,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,
@@ -372,6 +398,9 @@ export const SKUS = {
372
398
  "EPYC 1st Generation (Naples)": {
373
399
  tflops: 0.6,
374
400
  },
401
+ "Ryzen Zen 4 7000 (Threadripper)": {
402
+ tflops: 10.0,
403
+ },
375
404
  "Ryzen Zen4 7000 (Ryzen 9)": {
376
405
  tflops: 0.56,
377
406
  },
@@ -459,7 +488,7 @@ export const SKUS = {
459
488
  },
460
489
  "Apple M4 Pro": {
461
490
  tflops: 9.2,
462
- memory: [24, 48],
491
+ memory: [24, 48, 64],
463
492
  },
464
493
  "Apple M4 Max": {
465
494
  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;
@@ -452,6 +465,12 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
452
465
  countDownloads: string;
453
466
  snippets: () => string[];
454
467
  };
468
+ mitie: {
469
+ prettyLabel: string;
470
+ repoName: string;
471
+ repoUrl: string;
472
+ countDownloads: string;
473
+ };
455
474
  "ml-agents": {
456
475
  prettyLabel: string;
457
476
  repoName: string;
@@ -750,6 +769,11 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
750
769
  repoUrl: string;
751
770
  snippets: (model: ModelData) => string[];
752
771
  };
772
+ tabpfn: {
773
+ prettyLabel: string;
774
+ repoName: string;
775
+ repoUrl: string;
776
+ };
753
777
  "tic-clip": {
754
778
  prettyLabel: string;
755
779
  repoName: string;
@@ -795,6 +819,15 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
795
819
  repoUrl: string;
796
820
  countDownloads: string;
797
821
  };
822
+ ultralytics: {
823
+ prettyLabel: string;
824
+ repoName: string;
825
+ repoUrl: string;
826
+ docsUrl: string;
827
+ filter: false;
828
+ countDownloads: string;
829
+ snippets: (model: ModelData) => string[];
830
+ };
798
831
  "unity-sentis": {
799
832
  prettyLabel: string;
800
833
  repoName: string;
@@ -824,20 +857,21 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
824
857
  docsUrl: string;
825
858
  snippets: (model: ModelData) => string[];
826
859
  };
827
- yolov10: {
860
+ whisperkit: {
828
861
  prettyLabel: string;
829
862
  repoName: string;
830
863
  repoUrl: string;
831
864
  docsUrl: string;
832
- snippets: (model: ModelData) => string[];
865
+ snippets: () => string[];
866
+ countDownloads: string;
833
867
  };
834
- whisperkit: {
868
+ yolov10: {
835
869
  prettyLabel: string;
836
870
  repoName: string;
837
871
  repoUrl: string;
838
872
  docsUrl: string;
839
- snippets: () => string[];
840
873
  countDownloads: string;
874
+ snippets: (model: ModelData) => string[];
841
875
  };
842
876
  "3dtopia-xl": {
843
877
  prettyLabel: string;
@@ -850,5 +884,5 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
850
884
  };
851
885
  export type ModelLibraryKey = keyof typeof MODEL_LIBRARIES_UI_ELEMENTS;
852
886
  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")[];
887
+ 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" | "mitie" | "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
888
  //# 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;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA+zBI,CAAC;AAE7C,MAAM,MAAM,eAAe,GAAG,MAAM,OAAO,2BAA2B,CAAC;AAEvE,eAAO,MAAM,sBAAsB,EAA+C,eAAe,EAAE,CAAC;AAEpG,eAAO,MAAM,8BAA8B,ygDAQ1B,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",
@@ -414,6 +427,12 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
414
427
  countDownloads: `path:"MeshAnything_350m.pth"`,
415
428
  snippets: snippets.mesh_anything,
416
429
  },
430
+ mitie: {
431
+ prettyLabel: "MITIE",
432
+ repoName: "MITIE",
433
+ repoUrl: "https://github.com/mit-nlp/MITIE",
434
+ countDownloads: `path_filename:"total_word_feature_extractor"`,
435
+ },
417
436
  "ml-agents": {
418
437
  prettyLabel: "ml-agents",
419
438
  repoName: "ml-agents",
@@ -712,6 +731,11 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
712
731
  repoUrl: "https://github.com/TensorSpeech/TensorFlowTTS",
713
732
  snippets: snippets.tensorflowtts,
714
733
  },
734
+ tabpfn: {
735
+ prettyLabel: "TabPFN",
736
+ repoName: "TabPFN",
737
+ repoUrl: "https://github.com/PriorLabs/TabPFN",
738
+ },
715
739
  "tic-clip": {
716
740
  prettyLabel: "TiC-CLIP",
717
741
  repoName: "TiC-CLIP",
@@ -751,12 +775,21 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
751
775
  snippets: snippets.transformersJS,
752
776
  filter: true,
753
777
  },
754
- "trellis": {
778
+ trellis: {
755
779
  prettyLabel: "Trellis",
756
780
  repoName: "Trellis",
757
781
  repoUrl: "https://github.com/microsoft/TRELLIS",
758
782
  countDownloads: `path_extension:"safetensors"`,
759
783
  },
784
+ ultralytics: {
785
+ prettyLabel: "ultralytics",
786
+ repoName: "ultralytics",
787
+ repoUrl: "https://github.com/ultralytics/ultralytics",
788
+ docsUrl: "https://github.com/ultralytics/ultralytics",
789
+ filter: false,
790
+ countDownloads: `path_extension:"pt"`,
791
+ snippets: snippets.ultralytics,
792
+ },
760
793
  "unity-sentis": {
761
794
  prettyLabel: "unity-sentis",
762
795
  repoName: "unity-sentis",
@@ -786,13 +819,6 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
786
819
  docsUrl: "https://github.com/jasonppy/VoiceCraft",
787
820
  snippets: snippets.voicecraft,
788
821
  },
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
822
  whisperkit: {
797
823
  prettyLabel: "WhisperKit",
798
824
  repoName: "WhisperKit",
@@ -801,6 +827,15 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
801
827
  snippets: snippets.whisperkit,
802
828
  countDownloads: `path_filename:"model" AND path_extension:"mil" AND _exists_:"path_prefix"`,
803
829
  },
830
+ yolov10: {
831
+ // YOLOv10 is a fork of ultraLytics. Code snippets and download count are the same but the repo is different.
832
+ prettyLabel: "YOLOv10",
833
+ repoName: "YOLOv10",
834
+ repoUrl: "https://github.com/THU-MIG/yolov10",
835
+ docsUrl: "https://github.com/THU-MIG/yolov10",
836
+ countDownloads: `path_extension:"pt"`,
837
+ snippets: snippets.ultralytics,
838
+ },
804
839
  "3dtopia-xl": {
805
840
  prettyLabel: "3DTopia-XL",
806
841
  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.15",
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
@@ -14,7 +14,7 @@ export const TFLOPS_THRESHOLD_EU_AI_ACT_MODEL_TRAINING_TOTAL = 10 ** 13;
14
14
 
15
15
  export interface HardwareSpec {
16
16
  /**
17
- * Approximate value, in FP16 whenever possible.
17
+ * Approximate value, in FP16 whenever possible for GPUs and FP32 for CPUs.
18
18
  * This is only approximate/theoretical and shouldn't be taken too seriously.
19
19
  * Currently the CPU values are from cpu-monkey.com
20
20
  * while the GPU values are from techpowerup.com
@@ -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,
@@ -394,6 +420,9 @@ export const SKUS = {
394
420
  "EPYC 1st Generation (Naples)": {
395
421
  tflops: 0.6,
396
422
  },
423
+ "Ryzen Zen 4 7000 (Threadripper)": {
424
+ tflops: 10.0,
425
+ },
397
426
  "Ryzen Zen4 7000 (Ryzen 9)": {
398
427
  tflops: 0.56,
399
428
  },
@@ -481,7 +510,7 @@ export const SKUS = {
481
510
  },
482
511
  "Apple M4 Pro": {
483
512
  tflops: 9.2,
484
- memory: [24, 48],
513
+ memory: [24, 48, 64],
485
514
  },
486
515
  "Apple M4 Max": {
487
516
  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",
@@ -458,6 +471,12 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
458
471
  countDownloads: `path:"MeshAnything_350m.pth"`,
459
472
  snippets: snippets.mesh_anything,
460
473
  },
474
+ mitie: {
475
+ prettyLabel: "MITIE",
476
+ repoName: "MITIE",
477
+ repoUrl: "https://github.com/mit-nlp/MITIE",
478
+ countDownloads: `path_filename:"total_word_feature_extractor"`,
479
+ },
461
480
  "ml-agents": {
462
481
  prettyLabel: "ml-agents",
463
482
  repoName: "ml-agents",
@@ -756,6 +775,11 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
756
775
  repoUrl: "https://github.com/TensorSpeech/TensorFlowTTS",
757
776
  snippets: snippets.tensorflowtts,
758
777
  },
778
+ tabpfn: {
779
+ prettyLabel: "TabPFN",
780
+ repoName: "TabPFN",
781
+ repoUrl: "https://github.com/PriorLabs/TabPFN",
782
+ },
759
783
  "tic-clip": {
760
784
  prettyLabel: "TiC-CLIP",
761
785
  repoName: "TiC-CLIP",
@@ -795,12 +819,21 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
795
819
  snippets: snippets.transformersJS,
796
820
  filter: true,
797
821
  },
798
- "trellis": {
822
+ trellis: {
799
823
  prettyLabel: "Trellis",
800
824
  repoName: "Trellis",
801
825
  repoUrl: "https://github.com/microsoft/TRELLIS",
802
826
  countDownloads: `path_extension:"safetensors"`,
803
827
  },
828
+ ultralytics: {
829
+ prettyLabel: "ultralytics",
830
+ repoName: "ultralytics",
831
+ repoUrl: "https://github.com/ultralytics/ultralytics",
832
+ docsUrl: "https://github.com/ultralytics/ultralytics",
833
+ filter: false,
834
+ countDownloads: `path_extension:"pt"`,
835
+ snippets: snippets.ultralytics,
836
+ },
804
837
  "unity-sentis": {
805
838
  prettyLabel: "unity-sentis",
806
839
  repoName: "unity-sentis",
@@ -830,13 +863,6 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
830
863
  docsUrl: "https://github.com/jasonppy/VoiceCraft",
831
864
  snippets: snippets.voicecraft,
832
865
  },
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
866
  whisperkit: {
841
867
  prettyLabel: "WhisperKit",
842
868
  repoName: "WhisperKit",
@@ -845,6 +871,15 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
845
871
  snippets: snippets.whisperkit,
846
872
  countDownloads: `path_filename:"model" AND path_extension:"mil" AND _exists_:"path_prefix"`,
847
873
  },
874
+ yolov10: {
875
+ // YOLOv10 is a fork of ultraLytics. Code snippets and download count are the same but the repo is different.
876
+ prettyLabel: "YOLOv10",
877
+ repoName: "YOLOv10",
878
+ repoUrl: "https://github.com/THU-MIG/yolov10",
879
+ docsUrl: "https://github.com/THU-MIG/yolov10",
880
+ countDownloads: `path_extension:"pt"`,
881
+ snippets: snippets.ultralytics,
882
+ },
848
883
  "3dtopia-xl": {
849
884
  prettyLabel: "3DTopia-XL",
850
885
  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