@huggingface/tasks 0.13.17 → 0.15.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (195) hide show
  1. package/dist/commonjs/index.d.ts +1 -0
  2. package/dist/commonjs/index.d.ts.map +1 -1
  3. package/dist/commonjs/index.js +1 -0
  4. package/dist/commonjs/inference-providers.d.ts +10 -0
  5. package/dist/commonjs/inference-providers.d.ts.map +1 -0
  6. package/dist/commonjs/inference-providers.js +16 -0
  7. package/dist/commonjs/model-libraries-snippets.d.ts +2 -2
  8. package/dist/commonjs/model-libraries-snippets.d.ts.map +1 -1
  9. package/dist/commonjs/model-libraries-snippets.js +2 -2
  10. package/dist/commonjs/model-libraries.d.ts +2 -2
  11. package/dist/commonjs/snippets/curl.d.ts +8 -8
  12. package/dist/commonjs/snippets/curl.d.ts.map +1 -1
  13. package/dist/commonjs/snippets/curl.js +58 -30
  14. package/dist/commonjs/snippets/js.d.ts +11 -10
  15. package/dist/commonjs/snippets/js.d.ts.map +1 -1
  16. package/dist/commonjs/snippets/js.js +162 -53
  17. package/dist/commonjs/snippets/python.d.ts +12 -12
  18. package/dist/commonjs/snippets/python.d.ts.map +1 -1
  19. package/dist/commonjs/snippets/python.js +141 -71
  20. package/dist/commonjs/snippets/types.d.ts +1 -1
  21. package/dist/commonjs/snippets/types.d.ts.map +1 -1
  22. package/dist/commonjs/tasks/audio-classification/inference.d.ts +1 -1
  23. package/dist/commonjs/tasks/audio-classification/inference.d.ts.map +1 -1
  24. package/dist/commonjs/tasks/automatic-speech-recognition/inference.d.ts +1 -1
  25. package/dist/commonjs/tasks/automatic-speech-recognition/inference.d.ts.map +1 -1
  26. package/dist/commonjs/tasks/automatic-speech-recognition/inference.js +0 -5
  27. package/dist/commonjs/tasks/chat-completion/inference.d.ts.map +1 -1
  28. package/dist/commonjs/tasks/chat-completion/inference.js +0 -5
  29. package/dist/commonjs/tasks/depth-estimation/inference.d.ts.map +1 -1
  30. package/dist/commonjs/tasks/depth-estimation/inference.js +0 -5
  31. package/dist/commonjs/tasks/feature-extraction/inference.d.ts.map +1 -1
  32. package/dist/commonjs/tasks/feature-extraction/inference.js +0 -5
  33. package/dist/commonjs/tasks/image-classification/inference.d.ts +1 -1
  34. package/dist/commonjs/tasks/image-classification/inference.d.ts.map +1 -1
  35. package/dist/commonjs/tasks/image-segmentation/inference.d.ts +1 -1
  36. package/dist/commonjs/tasks/image-segmentation/inference.d.ts.map +1 -1
  37. package/dist/commonjs/tasks/image-to-image/inference.d.ts +3 -3
  38. package/dist/commonjs/tasks/image-to-image/inference.d.ts.map +1 -1
  39. package/dist/commonjs/tasks/image-to-image/inference.js +0 -5
  40. package/dist/commonjs/tasks/image-to-text/inference.d.ts +1 -1
  41. package/dist/commonjs/tasks/image-to-text/inference.d.ts.map +1 -1
  42. package/dist/commonjs/tasks/image-to-text/inference.js +0 -5
  43. package/dist/commonjs/tasks/index.d.ts +1 -0
  44. package/dist/commonjs/tasks/index.d.ts.map +1 -1
  45. package/dist/commonjs/tasks/object-detection/inference.d.ts +1 -1
  46. package/dist/commonjs/tasks/object-detection/inference.d.ts.map +1 -1
  47. package/dist/commonjs/tasks/sentence-similarity/inference.d.ts.map +1 -1
  48. package/dist/commonjs/tasks/sentence-similarity/inference.js +0 -5
  49. package/dist/commonjs/tasks/summarization/inference.d.ts.map +1 -1
  50. package/dist/commonjs/tasks/summarization/inference.js +0 -5
  51. package/dist/commonjs/tasks/text-generation/inference.d.ts.map +1 -1
  52. package/dist/commonjs/tasks/text-generation/inference.js +0 -5
  53. package/dist/commonjs/tasks/text-to-audio/inference.d.ts +14 -15
  54. package/dist/commonjs/tasks/text-to-audio/inference.d.ts.map +1 -1
  55. package/dist/commonjs/tasks/text-to-audio/inference.js +0 -5
  56. package/dist/commonjs/tasks/text-to-image/inference.d.ts +2 -2
  57. package/dist/commonjs/tasks/text-to-image/inference.d.ts.map +1 -1
  58. package/dist/commonjs/tasks/text-to-image/inference.js +0 -5
  59. package/dist/commonjs/tasks/text-to-speech/inference.d.ts +14 -17
  60. package/dist/commonjs/tasks/text-to-speech/inference.d.ts.map +1 -1
  61. package/dist/commonjs/tasks/text-to-speech/inference.js +0 -5
  62. package/dist/commonjs/tasks/text-to-video/inference.d.ts +58 -0
  63. package/dist/commonjs/tasks/text-to-video/inference.d.ts.map +1 -0
  64. package/dist/commonjs/tasks/text-to-video/inference.js +2 -0
  65. package/dist/commonjs/tasks/text2text-generation/inference.d.ts.map +1 -1
  66. package/dist/commonjs/tasks/text2text-generation/inference.js +0 -5
  67. package/dist/commonjs/tasks/translation/inference.d.ts.map +1 -1
  68. package/dist/commonjs/tasks/translation/inference.js +0 -5
  69. package/dist/commonjs/tasks/visual-question-answering/inference.d.ts +1 -1
  70. package/dist/commonjs/tasks/visual-question-answering/inference.d.ts.map +1 -1
  71. package/dist/commonjs/tasks/zero-shot-image-classification/inference.d.ts +1 -1
  72. package/dist/commonjs/tasks/zero-shot-image-classification/inference.d.ts.map +1 -1
  73. package/dist/commonjs/tasks/zero-shot-object-detection/inference.d.ts +1 -1
  74. package/dist/commonjs/tasks/zero-shot-object-detection/inference.d.ts.map +1 -1
  75. package/dist/esm/index.d.ts +1 -0
  76. package/dist/esm/index.d.ts.map +1 -1
  77. package/dist/esm/index.js +1 -0
  78. package/dist/esm/inference-providers.d.ts +10 -0
  79. package/dist/esm/inference-providers.d.ts.map +1 -0
  80. package/dist/esm/inference-providers.js +12 -0
  81. package/dist/esm/model-libraries-snippets.d.ts +2 -2
  82. package/dist/esm/model-libraries-snippets.d.ts.map +1 -1
  83. package/dist/esm/model-libraries-snippets.js +2 -2
  84. package/dist/esm/model-libraries.d.ts +2 -2
  85. package/dist/esm/snippets/curl.d.ts +8 -8
  86. package/dist/esm/snippets/curl.d.ts.map +1 -1
  87. package/dist/esm/snippets/curl.js +58 -29
  88. package/dist/esm/snippets/js.d.ts +11 -10
  89. package/dist/esm/snippets/js.d.ts.map +1 -1
  90. package/dist/esm/snippets/js.js +159 -50
  91. package/dist/esm/snippets/python.d.ts +12 -12
  92. package/dist/esm/snippets/python.d.ts.map +1 -1
  93. package/dist/esm/snippets/python.js +140 -69
  94. package/dist/esm/snippets/types.d.ts +1 -1
  95. package/dist/esm/snippets/types.d.ts.map +1 -1
  96. package/dist/esm/tasks/audio-classification/inference.d.ts +1 -1
  97. package/dist/esm/tasks/audio-classification/inference.d.ts.map +1 -1
  98. package/dist/esm/tasks/automatic-speech-recognition/inference.d.ts +1 -1
  99. package/dist/esm/tasks/automatic-speech-recognition/inference.d.ts.map +1 -1
  100. package/dist/esm/tasks/automatic-speech-recognition/inference.js +0 -5
  101. package/dist/esm/tasks/chat-completion/inference.d.ts.map +1 -1
  102. package/dist/esm/tasks/chat-completion/inference.js +0 -5
  103. package/dist/esm/tasks/depth-estimation/inference.d.ts.map +1 -1
  104. package/dist/esm/tasks/depth-estimation/inference.js +0 -5
  105. package/dist/esm/tasks/feature-extraction/inference.d.ts.map +1 -1
  106. package/dist/esm/tasks/feature-extraction/inference.js +0 -5
  107. package/dist/esm/tasks/image-classification/inference.d.ts +1 -1
  108. package/dist/esm/tasks/image-classification/inference.d.ts.map +1 -1
  109. package/dist/esm/tasks/image-segmentation/inference.d.ts +1 -1
  110. package/dist/esm/tasks/image-segmentation/inference.d.ts.map +1 -1
  111. package/dist/esm/tasks/image-to-image/inference.d.ts +3 -3
  112. package/dist/esm/tasks/image-to-image/inference.d.ts.map +1 -1
  113. package/dist/esm/tasks/image-to-image/inference.js +0 -5
  114. package/dist/esm/tasks/image-to-text/inference.d.ts +1 -1
  115. package/dist/esm/tasks/image-to-text/inference.d.ts.map +1 -1
  116. package/dist/esm/tasks/image-to-text/inference.js +0 -5
  117. package/dist/esm/tasks/index.d.ts +1 -0
  118. package/dist/esm/tasks/index.d.ts.map +1 -1
  119. package/dist/esm/tasks/object-detection/inference.d.ts +1 -1
  120. package/dist/esm/tasks/object-detection/inference.d.ts.map +1 -1
  121. package/dist/esm/tasks/sentence-similarity/inference.d.ts.map +1 -1
  122. package/dist/esm/tasks/sentence-similarity/inference.js +0 -5
  123. package/dist/esm/tasks/summarization/inference.d.ts.map +1 -1
  124. package/dist/esm/tasks/summarization/inference.js +0 -5
  125. package/dist/esm/tasks/text-generation/inference.d.ts.map +1 -1
  126. package/dist/esm/tasks/text-generation/inference.js +0 -5
  127. package/dist/esm/tasks/text-to-audio/inference.d.ts +14 -15
  128. package/dist/esm/tasks/text-to-audio/inference.d.ts.map +1 -1
  129. package/dist/esm/tasks/text-to-audio/inference.js +0 -5
  130. package/dist/esm/tasks/text-to-image/inference.d.ts +2 -2
  131. package/dist/esm/tasks/text-to-image/inference.d.ts.map +1 -1
  132. package/dist/esm/tasks/text-to-image/inference.js +0 -5
  133. package/dist/esm/tasks/text-to-speech/inference.d.ts +14 -17
  134. package/dist/esm/tasks/text-to-speech/inference.d.ts.map +1 -1
  135. package/dist/esm/tasks/text-to-speech/inference.js +0 -5
  136. package/dist/esm/tasks/text-to-video/inference.d.ts +58 -0
  137. package/dist/esm/tasks/text-to-video/inference.d.ts.map +1 -0
  138. package/dist/esm/tasks/text-to-video/inference.js +1 -0
  139. package/dist/esm/tasks/text2text-generation/inference.d.ts.map +1 -1
  140. package/dist/esm/tasks/text2text-generation/inference.js +0 -5
  141. package/dist/esm/tasks/translation/inference.d.ts.map +1 -1
  142. package/dist/esm/tasks/translation/inference.js +0 -5
  143. package/dist/esm/tasks/visual-question-answering/inference.d.ts +1 -1
  144. package/dist/esm/tasks/visual-question-answering/inference.d.ts.map +1 -1
  145. package/dist/esm/tasks/zero-shot-image-classification/inference.d.ts +1 -1
  146. package/dist/esm/tasks/zero-shot-image-classification/inference.d.ts.map +1 -1
  147. package/dist/esm/tasks/zero-shot-object-detection/inference.d.ts +1 -1
  148. package/dist/esm/tasks/zero-shot-object-detection/inference.d.ts.map +1 -1
  149. package/package.json +1 -1
  150. package/src/index.ts +2 -0
  151. package/src/inference-providers.ts +16 -0
  152. package/src/model-libraries-snippets.ts +3 -3
  153. package/src/snippets/curl.ts +72 -23
  154. package/src/snippets/js.ts +189 -56
  155. package/src/snippets/python.ts +154 -75
  156. package/src/snippets/types.ts +1 -1
  157. package/src/tasks/audio-classification/inference.ts +1 -1
  158. package/src/tasks/audio-classification/spec/input.json +2 -1
  159. package/src/tasks/automatic-speech-recognition/inference.ts +1 -7
  160. package/src/tasks/automatic-speech-recognition/spec/input.json +2 -1
  161. package/src/tasks/chat-completion/inference.ts +0 -33
  162. package/src/tasks/depth-estimation/inference.ts +3 -3
  163. package/src/tasks/document-question-answering/spec/input.json +2 -1
  164. package/src/tasks/feature-extraction/inference.ts +0 -3
  165. package/src/tasks/image-classification/inference.ts +1 -1
  166. package/src/tasks/image-classification/spec/input.json +2 -1
  167. package/src/tasks/image-segmentation/inference.ts +1 -1
  168. package/src/tasks/image-segmentation/spec/input.json +2 -1
  169. package/src/tasks/image-to-image/inference.ts +3 -7
  170. package/src/tasks/image-to-image/spec/input.json +4 -6
  171. package/src/tasks/image-to-text/inference.ts +1 -6
  172. package/src/tasks/image-to-text/spec/input.json +2 -1
  173. package/src/tasks/index.ts +1 -0
  174. package/src/tasks/object-detection/inference.ts +1 -1
  175. package/src/tasks/object-detection/spec/input.json +2 -1
  176. package/src/tasks/sentence-similarity/inference.ts +3 -4
  177. package/src/tasks/summarization/inference.ts +3 -5
  178. package/src/tasks/text-generation/inference.ts +0 -13
  179. package/src/tasks/text-to-audio/inference.ts +14 -20
  180. package/src/tasks/text-to-audio/spec/output.json +3 -2
  181. package/src/tasks/text-to-image/inference.ts +2 -6
  182. package/src/tasks/text-to-image/spec/input.json +2 -5
  183. package/src/tasks/text-to-speech/inference.ts +14 -22
  184. package/src/tasks/text-to-speech/spec/output.json +13 -2
  185. package/src/tasks/text-to-video/inference.ts +57 -0
  186. package/src/tasks/text-to-video/spec/input.json +49 -0
  187. package/src/tasks/text-to-video/spec/output.json +13 -0
  188. package/src/tasks/text2text-generation/inference.ts +3 -5
  189. package/src/tasks/translation/inference.ts +3 -5
  190. package/src/tasks/visual-question-answering/inference.ts +1 -1
  191. package/src/tasks/visual-question-answering/spec/input.json +4 -2
  192. package/src/tasks/zero-shot-image-classification/inference.ts +1 -1
  193. package/src/tasks/zero-shot-image-classification/spec/input.json +2 -1
  194. package/src/tasks/zero-shot-object-detection/inference.ts +1 -1
  195. package/src/tasks/zero-shot-object-detection/spec/input.json +2 -1
@@ -1 +1 @@
1
- {"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../../src/tasks/text-generation/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;;GAMG;AACH,MAAM,WAAW,mBAAmB;IACnC,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,CAAC,EAAE,qCAAqC,CAAC;IACnD,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED,MAAM,WAAW,qCAAqC;IACrD;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAChC;;OAEG;IACH,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB;;OAEG;IACH,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB;;;;OAIG;IACH,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,OAAO,CAAC,EAAE,8BAA8B,CAAC;IACzC;;OAEG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;;OAGG;IACH,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B;;OAEG;IACH,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B;;OAEG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB;;OAEG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB;;;;OAIG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB;;;OAGG;IACH,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED,MAAM,WAAW,8BAA8B;IAC9C,IAAI,EAAE,IAAI,CAAC;IACX;;;;;OAKG;IACH,KAAK,EAAE,OAAO,CAAC;IACf,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED,MAAM,MAAM,IAAI,GAAG,MAAM,GAAG,OAAO,CAAC;AAEpC;;;;;;GAMG;AACH,MAAM,WAAW,oBAAoB;IACpC,OAAO,CAAC,EAAE,2BAA2B,CAAC;IACtC,cAAc,EAAE,MAAM,CAAC;IACvB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED,MAAM,WAAW,2BAA2B;IAC3C,iBAAiB,CAAC,EAAE,kCAAkC,EAAE,CAAC;IACzD,aAAa,EAAE,gCAAgC,CAAC;IAChD,gBAAgB,EAAE,MAAM,CAAC;IACzB,OAAO,EAAE,gCAAgC,EAAE,CAAC;IAC5C,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,yBAAyB,EAAE,CAAC;IACpC,UAAU,CAAC,EAAE,KAAK,CAAC,yBAAyB,EAAE,CAAC,CAAC;IAChD,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED,MAAM,WAAW,kCAAkC;IAClD,aAAa,EAAE,gCAAgC,CAAC;IAChD,cAAc,EAAE,MAAM,CAAC;IACvB,gBAAgB,EAAE,MAAM,CAAC;IACzB,OAAO,EAAE,gCAAgC,EAAE,CAAC;IAC5C,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,yBAAyB,EAAE,CAAC;IACpC,UAAU,CAAC,EAAE,KAAK,CAAC,yBAAyB,EAAE,CAAC,CAAC;IAChD,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED,MAAM,MAAM,gCAAgC,GAAG,QAAQ,GAAG,WAAW,GAAG,eAAe,CAAC;AAExF,MAAM,WAAW,gCAAgC;IAChD,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED,MAAM,WAAW,yBAAyB;IACzC,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,OAAO,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED;;;;;;GAMG;AACH,MAAM,WAAW,0BAA0B;IAC1C,OAAO,CAAC,EAAE,uCAAuC,CAAC;IAClD,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,+BAA+B,CAAC;IACvC,UAAU,CAAC,EAAE,+BAA+B,EAAE,CAAC;IAC/C,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED,MAAM,WAAW,uCAAuC;IACvD,aAAa,EAAE,gCAAgC,CAAC;IAChD,gBAAgB,EAAE,MAAM,CAAC;IACzB,YAAY,EAAE,MAAM,CAAC;IACrB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED,MAAM,WAAW,+BAA+B;IAC/C,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,OAAO,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}
1
+ {"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../../src/tasks/text-generation/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH;;;;;;GAMG;AACH,MAAM,WAAW,mBAAmB;IACnC,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,CAAC,EAAE,qCAAqC,CAAC;IACnD,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD,MAAM,WAAW,qCAAqC;IACrD;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAChC;;OAEG;IACH,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB;;OAEG;IACH,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB;;;;OAIG;IACH,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,OAAO,CAAC,EAAE,8BAA8B,CAAC;IACzC;;OAEG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;;OAGG;IACH,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B;;OAEG;IACH,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B;;OAEG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB;;OAEG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB;;;;OAIG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB;;;OAGG;IACH,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD,MAAM,WAAW,8BAA8B;IAC9C,IAAI,EAAE,IAAI,CAAC;IACX;;;;;OAKG;IACH,KAAK,EAAE,OAAO,CAAC;IACf,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD,MAAM,MAAM,IAAI,GAAG,MAAM,GAAG,OAAO,CAAC;AACpC;;;;;;GAMG;AACH,MAAM,WAAW,oBAAoB;IACpC,OAAO,CAAC,EAAE,2BAA2B,CAAC;IACtC,cAAc,EAAE,MAAM,CAAC;IACvB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD,MAAM,WAAW,2BAA2B;IAC3C,iBAAiB,CAAC,EAAE,kCAAkC,EAAE,CAAC;IACzD,aAAa,EAAE,gCAAgC,CAAC;IAChD,gBAAgB,EAAE,MAAM,CAAC;IACzB,OAAO,EAAE,gCAAgC,EAAE,CAAC;IAC5C,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,yBAAyB,EAAE,CAAC;IACpC,UAAU,CAAC,EAAE,KAAK,CAAC,yBAAyB,EAAE,CAAC,CAAC;IAChD,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD,MAAM,WAAW,kCAAkC;IAClD,aAAa,EAAE,gCAAgC,CAAC;IAChD,cAAc,EAAE,MAAM,CAAC;IACvB,gBAAgB,EAAE,MAAM,CAAC;IACzB,OAAO,EAAE,gCAAgC,EAAE,CAAC;IAC5C,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,yBAAyB,EAAE,CAAC;IACpC,UAAU,CAAC,EAAE,KAAK,CAAC,yBAAyB,EAAE,CAAC,CAAC;IAChD,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD,MAAM,MAAM,gCAAgC,GAAG,QAAQ,GAAG,WAAW,GAAG,eAAe,CAAC;AACxF,MAAM,WAAW,gCAAgC;IAChD,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD,MAAM,WAAW,yBAAyB;IACzC,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,OAAO,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;;;;;GAMG;AACH,MAAM,WAAW,0BAA0B;IAC1C,OAAO,CAAC,EAAE,uCAAuC,CAAC;IAClD,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,+BAA+B,CAAC;IACvC,UAAU,CAAC,EAAE,+BAA+B,EAAE,CAAC;IAC/C,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD,MAAM,WAAW,uCAAuC;IACvD,aAAa,EAAE,gCAAgC,CAAC;IAChD,gBAAgB,EAAE,MAAM,CAAC;IACzB,YAAY,EAAE,MAAM,CAAC;IACrB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD,MAAM,WAAW,+BAA+B;IAC/C,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,OAAO,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}
@@ -1,6 +1 @@
1
- /**
2
- * Inference code generated from the JSON schema spec in ./spec
3
- *
4
- * Using src/scripts/inference-codegen
5
- */
6
1
  export {};
@@ -1,3 +1,17 @@
1
+ /**
2
+ * Outputs of inference for the Text To Audio task
3
+ */
4
+ export interface TextToAudioOutput {
5
+ /**
6
+ * The generated audio waveform.
7
+ */
8
+ audio: Blob;
9
+ /**
10
+ * The sampling rate of the generated audio waveform.
11
+ */
12
+ sampling_rate: number;
13
+ [property: string]: unknown;
14
+ }
1
15
  /**
2
16
  * Inference code generated from the JSON schema spec in ./spec
3
17
  *
@@ -117,19 +131,4 @@ export interface GenerationParameters {
117
131
  * Controls the stopping condition for beam-based methods.
118
132
  */
119
133
  export type EarlyStoppingUnion = boolean | "never";
120
- /**
121
- * Outputs of inference for the Text To Audio task
122
- */
123
- export interface TextToAudioOutput {
124
- /**
125
- * The generated audio waveform.
126
- */
127
- audio: unknown;
128
- samplingRate: unknown;
129
- /**
130
- * The sampling rate of the generated audio waveform.
131
- */
132
- sampling_rate?: number;
133
- [property: string]: unknown;
134
- }
135
134
  //# sourceMappingURL=inference.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../../src/tasks/text-to-audio/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,UAAU,CAAC,EAAE,qBAAqB,CAAC;IACnC,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACrC;;OAEG;IACH,qBAAqB,CAAC,EAAE,oBAAoB,CAAC;IAC7C,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACpC;;OAEG;IACH,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB;;OAEG;IACH,cAAc,CAAC,EAAE,kBAAkB,CAAC;IACpC;;;;;OAKG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;;;;;;;OAQG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;;OAGG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB;;OAEG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB;;;OAGG;IACH,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB;;OAEG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;;OAGG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;;;;;OAMG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB;;OAEG;IACH,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED;;GAEG;AACH,MAAM,MAAM,kBAAkB,GAAG,OAAO,GAAG,OAAO,CAAC;AAEnD;;GAEG;AACH,MAAM,WAAW,iBAAiB;IACjC;;OAEG;IACH,KAAK,EAAE,OAAO,CAAC;IACf,YAAY,EAAE,OAAO,CAAC;IACtB;;OAEG;IACH,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}
1
+ {"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../../src/tasks/text-to-audio/inference.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,WAAW,iBAAiB;IACjC;;OAEG;IACH,KAAK,EAAE,IAAI,CAAC;IACZ;;OAEG;IACH,aAAa,EAAE,MAAM,CAAC;IACtB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;;;GAIG;AACH;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,UAAU,CAAC,EAAE,qBAAqB,CAAC;IACnC,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACrC;;OAEG;IACH,qBAAqB,CAAC,EAAE,oBAAoB,CAAC;IAC7C,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACpC;;OAEG;IACH,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB;;OAEG;IACH,cAAc,CAAC,EAAE,kBAAkB,CAAC;IACpC;;;;;OAKG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;;;;;;;OAQG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;;OAGG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB;;OAEG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB;;;OAGG;IACH,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB;;OAEG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;;OAGG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;;;;;OAMG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB;;OAEG;IACH,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,MAAM,kBAAkB,GAAG,OAAO,GAAG,OAAO,CAAC"}
@@ -1,6 +1 @@
1
- /**
2
- * Inference code generated from the JSON schema spec in ./spec
3
- *
4
- * Using src/scripts/inference-codegen
5
- */
6
1
  export {};
@@ -27,9 +27,9 @@ export interface TextToImageParameters {
27
27
  */
28
28
  guidance_scale?: number;
29
29
  /**
30
- * One or several prompt to guide what NOT to include in image generation.
30
+ * One prompt to guide what NOT to include in image generation.
31
31
  */
32
- negative_prompt?: string[];
32
+ negative_prompt?: string;
33
33
  /**
34
34
  * The number of denoising steps. More denoising steps usually lead to a higher quality
35
35
  * image at the expense of slower inference.
@@ -1 +1 @@
1
- {"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../../src/tasks/text-to-image/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,UAAU,CAAC,EAAE,qBAAqB,CAAC;IACnC,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACrC;;;OAGG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;OAEG;IACH,eAAe,CAAC,EAAE,MAAM,EAAE,CAAC;IAC3B;;;OAGG;IACH,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B;;OAEG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB;;OAEG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,WAAW,CAAC,EAAE,UAAU,CAAC;IACzB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED;;GAEG;AACH,MAAM,WAAW,UAAU;IAC1B,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IACjC;;OAEG;IACH,KAAK,EAAE,OAAO,CAAC;IACf,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}
1
+ {"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../../src/tasks/text-to-image/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,UAAU,CAAC,EAAE,qBAAqB,CAAC;IACnC,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACrC;;;OAGG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;OAEG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB;;;OAGG;IACH,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B;;OAEG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB;;OAEG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,WAAW,CAAC,EAAE,UAAU,CAAC;IACzB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,WAAW,UAAU;IAC1B,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,WAAW,iBAAiB;IACjC;;OAEG;IACH,KAAK,EAAE,OAAO,CAAC;IACf,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}
@@ -1,6 +1 @@
1
- /**
2
- * Inference code generated from the JSON schema spec in ./spec
3
- *
4
- * Using src/scripts/inference-codegen
5
- */
6
1
  export {};
@@ -1,3 +1,17 @@
1
+ /**
2
+ * Outputs of inference for the Text To Speech task
3
+ */
4
+ export interface TextToSpeechOutput {
5
+ /**
6
+ * The generated audio
7
+ */
8
+ audio: Blob;
9
+ /**
10
+ * The sampling rate of the generated audio waveform.
11
+ */
12
+ sampling_rate?: number;
13
+ [property: string]: unknown;
14
+ }
1
15
  /**
2
16
  * Inference code generated from the JSON schema spec in ./spec
3
17
  *
@@ -117,21 +131,4 @@ export interface GenerationParameters {
117
131
  * Controls the stopping condition for beam-based methods.
118
132
  */
119
133
  export type EarlyStoppingUnion = boolean | "never";
120
- /**
121
- * Outputs for Text to Speech inference
122
- *
123
- * Outputs of inference for the Text To Audio task
124
- */
125
- export interface TextToSpeechOutput {
126
- /**
127
- * The generated audio waveform.
128
- */
129
- audio: unknown;
130
- samplingRate: unknown;
131
- /**
132
- * The sampling rate of the generated audio waveform.
133
- */
134
- sampling_rate?: number;
135
- [property: string]: unknown;
136
- }
137
134
  //# sourceMappingURL=inference.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../../src/tasks/text-to-speech/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;GAEG;AACH,MAAM,WAAW,iBAAiB;IACjC;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,UAAU,CAAC,EAAE,sBAAsB,CAAC;IACpC,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACtC;;OAEG;IACH,qBAAqB,CAAC,EAAE,oBAAoB,CAAC;IAC7C,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACpC;;OAEG;IACH,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB;;OAEG;IACH,cAAc,CAAC,EAAE,kBAAkB,CAAC;IACpC;;;;;OAKG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;;;;;;;OAQG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;;OAGG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB;;OAEG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB;;;OAGG;IACH,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB;;OAEG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;;OAGG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;;;;;OAMG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB;;OAEG;IACH,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED;;GAEG;AACH,MAAM,MAAM,kBAAkB,GAAG,OAAO,GAAG,OAAO,CAAC;AAEnD;;;;GAIG;AACH,MAAM,WAAW,kBAAkB;IAClC;;OAEG;IACH,KAAK,EAAE,OAAO,CAAC;IACf,YAAY,EAAE,OAAO,CAAC;IACtB;;OAEG;IACH,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}
1
+ {"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../../src/tasks/text-to-speech/inference.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,WAAW,kBAAkB;IAClC;;OAEG;IACH,KAAK,EAAE,IAAI,CAAC;IACZ;;OAEG;IACH,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;;;GAIG;AACH;;GAEG;AACH,MAAM,WAAW,iBAAiB;IACjC;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,UAAU,CAAC,EAAE,sBAAsB,CAAC;IACpC,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACtC;;OAEG;IACH,qBAAqB,CAAC,EAAE,oBAAoB,CAAC;IAC7C,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACpC;;OAEG;IACH,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB;;OAEG;IACH,cAAc,CAAC,EAAE,kBAAkB,CAAC;IACpC;;;;;OAKG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;;;;;;;OAQG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;;OAGG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB;;OAEG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB;;;OAGG;IACH,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB;;OAEG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;;OAGG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;;;;;OAMG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB;;OAEG;IACH,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,MAAM,kBAAkB,GAAG,OAAO,GAAG,OAAO,CAAC"}
@@ -1,6 +1 @@
1
- /**
2
- * Inference code generated from the JSON schema spec in ./spec
3
- *
4
- * Using src/scripts/inference-codegen
5
- */
6
1
  export {};
@@ -0,0 +1,58 @@
1
+ /**
2
+ * Inference code generated from the JSON schema spec in ./spec
3
+ *
4
+ * Using src/scripts/inference-codegen
5
+ */
6
+ /**
7
+ * Inputs for Text To Video inference
8
+ */
9
+ export interface TextToVideoInput {
10
+ /**
11
+ * The input text data (sometimes called "prompt")
12
+ */
13
+ inputs: string;
14
+ /**
15
+ * Additional inference parameters for Text To Video
16
+ */
17
+ parameters?: TextToVideoParameters;
18
+ [property: string]: unknown;
19
+ }
20
+ /**
21
+ * Additional inference parameters for Text To Video
22
+ */
23
+ export interface TextToVideoParameters {
24
+ /**
25
+ * A higher guidance scale value encourages the model to generate images closely linked to
26
+ * the text prompt, but values too high may cause saturation and other artifacts.
27
+ */
28
+ guidance_scale?: number;
29
+ /**
30
+ * One or several prompt to guide what NOT to include in image generation.
31
+ */
32
+ negative_prompt?: string[];
33
+ /**
34
+ * The num_frames parameter determines how many video frames are generated.
35
+ */
36
+ num_frames?: number;
37
+ /**
38
+ * The number of denoising steps. More denoising steps usually lead to a higher quality
39
+ * image at the expense of slower inference.
40
+ */
41
+ num_inference_steps?: number;
42
+ /**
43
+ * Seed for the random number generator.
44
+ */
45
+ seed?: number;
46
+ [property: string]: unknown;
47
+ }
48
+ /**
49
+ * Outputs of inference for the Text To Video task
50
+ */
51
+ export interface TextToVideoOutput {
52
+ /**
53
+ * The generated video returned as raw bytes in the payload.
54
+ */
55
+ video: unknown;
56
+ [property: string]: unknown;
57
+ }
58
+ //# sourceMappingURL=inference.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../../src/tasks/text-to-video/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,UAAU,CAAC,EAAE,qBAAqB,CAAC;IACnC,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACrC;;;OAGG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;OAEG;IACH,eAAe,CAAC,EAAE,MAAM,EAAE,CAAC;IAC3B;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;;OAGG;IACH,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B;;OAEG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,WAAW,iBAAiB;IACjC;;OAEG;IACH,KAAK,EAAE,OAAO,CAAC;IACf,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}
@@ -0,0 +1 @@
1
+ export {};
@@ -1 +1 @@
1
- {"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../../src/tasks/text2text-generation/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;GAEG;AACH,MAAM,WAAW,wBAAwB;IACxC;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,UAAU,CAAC,EAAE,6BAA6B,CAAC;IAC3C,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED;;GAEG;AACH,MAAM,WAAW,6BAA6B;IAC7C;;OAEG;IACH,4BAA4B,CAAC,EAAE,OAAO,CAAC;IACvC;;OAEG;IACH,mBAAmB,CAAC,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;KAAE,CAAC;IACjD;;OAEG;IACH,UAAU,CAAC,EAAE,qCAAqC,CAAC;IACnD,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED,MAAM,MAAM,qCAAqC,GAAG,iBAAiB,GAAG,eAAe,GAAG,YAAY,GAAG,aAAa,CAAC;AAEvH;;GAEG;AACH,MAAM,WAAW,yBAAyB;IACzC,aAAa,EAAE,OAAO,CAAC;IACvB;;OAEG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}
1
+ {"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../../src/tasks/text2text-generation/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH;;GAEG;AACH,MAAM,WAAW,wBAAwB;IACxC;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,UAAU,CAAC,EAAE,6BAA6B,CAAC;IAC3C,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,WAAW,6BAA6B;IAC7C;;OAEG;IACH,4BAA4B,CAAC,EAAE,OAAO,CAAC;IACvC;;OAEG;IACH,mBAAmB,CAAC,EAAE;QACrB,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;KACvB,CAAC;IACF;;OAEG;IACH,UAAU,CAAC,EAAE,qCAAqC,CAAC;IACnD,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD,MAAM,MAAM,qCAAqC,GAAG,iBAAiB,GAAG,eAAe,GAAG,YAAY,GAAG,aAAa,CAAC;AACvH;;GAEG;AACH,MAAM,WAAW,yBAAyB;IACzC,aAAa,EAAE,OAAO,CAAC;IACvB;;OAEG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}
@@ -1,6 +1 @@
1
- /**
2
- * Inference code generated from the JSON schema spec in ./spec
3
- *
4
- * Using src/scripts/inference-codegen
5
- */
6
1
  export {};
@@ -1 +1 @@
1
- {"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../../src/tasks/translation/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,UAAU,CAAC,EAAE,qBAAqB,CAAC;IACnC,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACrC;;OAEG;IACH,4BAA4B,CAAC,EAAE,OAAO,CAAC;IACvC;;OAEG;IACH,mBAAmB,CAAC,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;KAAE,CAAC;IACjD;;;OAGG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB;;;OAGG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB;;OAEG;IACH,UAAU,CAAC,EAAE,6BAA6B,CAAC;IAC3C,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED;;GAEG;AACH,MAAM,MAAM,6BAA6B,GAAG,iBAAiB,GAAG,eAAe,GAAG,YAAY,GAAG,aAAa,CAAC;AAE/G;;GAEG;AACH,MAAM,WAAW,iBAAiB;IACjC;;OAEG;IACH,gBAAgB,EAAE,MAAM,CAAC;IACzB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}
1
+ {"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../../src/tasks/translation/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,UAAU,CAAC,EAAE,qBAAqB,CAAC;IACnC,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACrC;;OAEG;IACH,4BAA4B,CAAC,EAAE,OAAO,CAAC;IACvC;;OAEG;IACH,mBAAmB,CAAC,EAAE;QACrB,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;KACvB,CAAC;IACF;;;OAGG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB;;;OAGG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB;;OAEG;IACH,UAAU,CAAC,EAAE,6BAA6B,CAAC;IAC3C,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,MAAM,6BAA6B,GAAG,iBAAiB,GAAG,eAAe,GAAG,YAAY,GAAG,aAAa,CAAC;AAC/G;;GAEG;AACH,MAAM,WAAW,iBAAiB;IACjC;;OAEG;IACH,gBAAgB,EAAE,MAAM,CAAC;IACzB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}
@@ -1,6 +1 @@
1
- /**
2
- * Inference code generated from the JSON schema spec in ./spec
3
- *
4
- * Using src/scripts/inference-codegen
5
- */
6
1
  export {};
@@ -28,7 +28,7 @@ export interface VisualQuestionAnsweringInputData {
28
28
  /**
29
29
  * The question to answer based on the image.
30
30
  */
31
- question: unknown;
31
+ question: string;
32
32
  [property: string]: unknown;
33
33
  }
34
34
  /**
@@ -1 +1 @@
1
- {"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../../src/tasks/visual-question-answering/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH;;GAEG;AACH,MAAM,WAAW,4BAA4B;IAC5C;;OAEG;IACH,MAAM,EAAE,gCAAgC,CAAC;IACzC;;OAEG;IACH,UAAU,CAAC,EAAE,iCAAiC,CAAC;IAC/C,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,WAAW,gCAAgC;IAChD;;OAEG;IACH,KAAK,EAAE,OAAO,CAAC;IACf;;OAEG;IACH,QAAQ,EAAE,OAAO,CAAC;IAClB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,WAAW,iCAAiC;IACjD;;;;OAIG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD,MAAM,MAAM,6BAA6B,GAAG,oCAAoC,EAAE,CAAC;AACnF;;GAEG;AACH,MAAM,WAAW,oCAAoC;IACpD;;OAEG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}
1
+ {"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../../src/tasks/visual-question-answering/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH;;GAEG;AACH,MAAM,WAAW,4BAA4B;IAC5C;;OAEG;IACH,MAAM,EAAE,gCAAgC,CAAC;IACzC;;OAEG;IACH,UAAU,CAAC,EAAE,iCAAiC,CAAC;IAC/C,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,WAAW,gCAAgC;IAChD;;OAEG;IACH,KAAK,EAAE,OAAO,CAAC;IACf;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,WAAW,iCAAiC;IACjD;;;;OAIG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD,MAAM,MAAM,6BAA6B,GAAG,oCAAoC,EAAE,CAAC;AACnF;;GAEG;AACH,MAAM,WAAW,oCAAoC;IACpD;;OAEG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}
@@ -10,7 +10,7 @@ export interface ZeroShotImageClassificationInput {
10
10
  /**
11
11
  * The input image data to classify as a base64-encoded string.
12
12
  */
13
- inputs: string;
13
+ inputs: Blob;
14
14
  /**
15
15
  * Additional inference parameters for Zero Shot Image Classification
16
16
  */
@@ -1 +1 @@
1
- {"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../../src/tasks/zero-shot-image-classification/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH;;GAEG;AACH,MAAM,WAAW,gCAAgC;IAChD;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,UAAU,EAAE,qCAAqC,CAAC;IAClD,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,WAAW,qCAAqC;IACrD;;OAEG;IACH,gBAAgB,EAAE,MAAM,EAAE,CAAC;IAC3B;;;OAGG;IACH,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD,MAAM,MAAM,iCAAiC,GAAG,wCAAwC,EAAE,CAAC;AAC3F;;GAEG;AACH,MAAM,WAAW,wCAAwC;IACxD;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}
1
+ {"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../../src/tasks/zero-shot-image-classification/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH;;GAEG;AACH,MAAM,WAAW,gCAAgC;IAChD;;OAEG;IACH,MAAM,EAAE,IAAI,CAAC;IACb;;OAEG;IACH,UAAU,EAAE,qCAAqC,CAAC;IAClD,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,WAAW,qCAAqC;IACrD;;OAEG;IACH,gBAAgB,EAAE,MAAM,EAAE,CAAC;IAC3B;;;OAGG;IACH,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD,MAAM,MAAM,iCAAiC,GAAG,wCAAwC,EAAE,CAAC;AAC3F;;GAEG;AACH,MAAM,WAAW,wCAAwC;IACxD;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}
@@ -10,7 +10,7 @@ export interface ZeroShotObjectDetectionInput {
10
10
  /**
11
11
  * The input image data as a base64-encoded string.
12
12
  */
13
- inputs: string;
13
+ inputs: Blob;
14
14
  /**
15
15
  * Additional inference parameters for Zero Shot Object Detection
16
16
  */
@@ -1 +1 @@
1
- {"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../../src/tasks/zero-shot-object-detection/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH;;GAEG;AACH,MAAM,WAAW,4BAA4B;IAC5C;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,UAAU,EAAE,iCAAiC,CAAC;IAC9C,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,WAAW,iCAAiC;IACjD;;OAEG;IACH,gBAAgB,EAAE,MAAM,EAAE,CAAC;IAC3B,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;;GAGG;AACH,MAAM,WAAW,WAAW;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD,MAAM,MAAM,6BAA6B,GAAG,oCAAoC,EAAE,CAAC;AACnF;;GAEG;AACH,MAAM,WAAW,oCAAoC;IACpD;;;OAGG;IACH,GAAG,EAAE,WAAW,CAAC;IACjB;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}
1
+ {"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../../src/tasks/zero-shot-object-detection/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH;;GAEG;AACH,MAAM,WAAW,4BAA4B;IAC5C;;OAEG;IACH,MAAM,EAAE,IAAI,CAAC;IACb;;OAEG;IACH,UAAU,EAAE,iCAAiC,CAAC;IAC9C,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,WAAW,iCAAiC;IACjD;;OAEG;IACH,gBAAgB,EAAE,MAAM,EAAE,CAAC;IAC3B,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;;GAGG;AACH,MAAM,WAAW,WAAW;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD,MAAM,MAAM,6BAA6B,GAAG,oCAAoC,EAAE,CAAC;AACnF;;GAEG;AACH,MAAM,WAAW,oCAAoC;IACpD;;;OAGG;IACH,GAAG,EAAE,WAAW,CAAC;IACjB;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}
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.17",
4
+ "version": "0.15.0",
5
5
  "description": "List of ML tasks for huggingface.co/tasks",
6
6
  "repository": "https://github.com/huggingface/huggingface.js.git",
7
7
  "publishConfig": {
package/src/index.ts CHANGED
@@ -58,3 +58,5 @@ export type { LocalApp, LocalAppKey, LocalAppSnippet } from "./local-apps.js";
58
58
 
59
59
  export { DATASET_LIBRARIES_UI_ELEMENTS } from "./dataset-libraries.js";
60
60
  export type { DatasetLibraryUiElement, DatasetLibraryKey } from "./dataset-libraries.js";
61
+
62
+ export * from "./inference-providers.js";
@@ -0,0 +1,16 @@
1
+ export const INFERENCE_PROVIDERS = ["hf-inference", "fal-ai", "replicate", "sambanova", "together"] as const;
2
+
3
+ export type InferenceProvider = (typeof INFERENCE_PROVIDERS)[number];
4
+
5
+ export const HF_HUB_INFERENCE_PROXY_TEMPLATE = `https://huggingface.co/api/inference-proxy/{{PROVIDER}}`;
6
+
7
+ /**
8
+ * URL to set as baseUrl in the OpenAI SDK.
9
+ *
10
+ * TODO(Expose this from HfInference in the future?)
11
+ */
12
+ export function openAIbaseUrl(provider: InferenceProvider): string {
13
+ return provider === "hf-inference"
14
+ ? "https://api-inference.huggingface.co/v1/"
15
+ : HF_HUB_INFERENCE_PROXY_TEMPLATE.replace("{{PROVIDER}}", provider);
16
+ }
@@ -95,7 +95,7 @@ export const bm25s = (model: ModelData): string[] => [
95
95
  retriever = BM25HF.load_from_hub("${model.id}")`,
96
96
  ];
97
97
 
98
- export const cxr_foundation = (model: ModelData): string[] => [
98
+ export const cxr_foundation = (): string[] => [
99
99
  `!git clone https://github.com/Google-Health/cxr-foundation.git
100
100
  import tensorflow as tf, sys, requests
101
101
  sys.path.append('cxr-foundation/python/')
@@ -191,7 +191,7 @@ focallength_px = prediction["focallength_px"]`;
191
191
  return [installSnippet, inferenceSnippet];
192
192
  };
193
193
 
194
- export const derm_foundation = (model: ModelData): string[] => [
194
+ export const derm_foundation = (): string[] => [
195
195
  `from huggingface_hub import from_pretrained_keras
196
196
  import tensorflow as tf, requests
197
197
 
@@ -211,7 +211,7 @@ input_tensor = tf.train.Example(
211
211
  loaded_model = from_pretrained_keras("google/derm-foundation")
212
212
  infer = loaded_model.signatures["serving_default"]
213
213
  print(infer(inputs=tf.constant([input_tensor])))`,
214
- ]
214
+ ];
215
215
 
216
216
  const diffusersDefaultPrompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k";
217
217
 
@@ -1,20 +1,35 @@
1
+ import { HF_HUB_INFERENCE_PROXY_TEMPLATE, type InferenceProvider } from "../inference-providers.js";
1
2
  import type { PipelineType } from "../pipelines.js";
2
3
  import type { ChatCompletionInputMessage, GenerationParameters } from "../tasks/index.js";
3
4
  import { stringifyGenerationConfig, stringifyMessages } from "./common.js";
4
5
  import { getModelInputSnippet } from "./inputs.js";
5
6
  import type { InferenceSnippet, ModelDataMinimal } from "./types.js";
6
7
 
7
- export const snippetBasic = (model: ModelDataMinimal, accessToken: string): InferenceSnippet => ({
8
- content: `curl https://api-inference.huggingface.co/models/${model.id} \\
8
+ export const snippetBasic = (
9
+ model: ModelDataMinimal,
10
+ accessToken: string,
11
+ provider: InferenceProvider
12
+ ): InferenceSnippet[] => {
13
+ if (provider !== "hf-inference") {
14
+ return [];
15
+ }
16
+ return [
17
+ {
18
+ client: "curl",
19
+ content: `\
20
+ curl https://api-inference.huggingface.co/models/${model.id} \\
9
21
  -X POST \\
10
22
  -d '{"inputs": ${getModelInputSnippet(model, true)}}' \\
11
23
  -H 'Content-Type: application/json' \\
12
24
  -H 'Authorization: Bearer ${accessToken || `{API_TOKEN}`}'`,
13
- });
25
+ },
26
+ ];
27
+ };
14
28
 
15
29
  export const snippetTextGeneration = (
16
30
  model: ModelDataMinimal,
17
31
  accessToken: string,
32
+ provider: InferenceProvider,
18
33
  opts?: {
19
34
  streaming?: boolean;
20
35
  messages?: ChatCompletionInputMessage[];
@@ -22,8 +37,13 @@ export const snippetTextGeneration = (
22
37
  max_tokens?: GenerationParameters["max_tokens"];
23
38
  top_p?: GenerationParameters["top_p"];
24
39
  }
25
- ): InferenceSnippet => {
40
+ ): InferenceSnippet[] => {
26
41
  if (model.tags.includes("conversational")) {
42
+ const baseUrl =
43
+ provider === "hf-inference"
44
+ ? `https://api-inference.huggingface.co/models/${model.id}/v1/chat/completions`
45
+ : HF_HUB_INFERENCE_PROXY_TEMPLATE.replace("{{PROVIDER}}", provider) + "/v1/chat/completions";
46
+
27
47
  // Conversational model detected, so we display a code snippet that features the Messages API
28
48
  const streaming = opts?.streaming ?? true;
29
49
  const exampleMessages = getModelInputSnippet(model) as ChatCompletionInputMessage[];
@@ -34,8 +54,10 @@ export const snippetTextGeneration = (
34
54
  max_tokens: opts?.max_tokens ?? 500,
35
55
  ...(opts?.top_p ? { top_p: opts.top_p } : undefined),
36
56
  };
37
- return {
38
- content: `curl 'https://api-inference.huggingface.co/models/${model.id}/v1/chat/completions' \\
57
+ return [
58
+ {
59
+ client: "curl",
60
+ content: `curl '${baseUrl}' \\
39
61
  -H 'Authorization: Bearer ${accessToken || `{API_TOKEN}`}' \\
40
62
  -H 'Content-Type: application/json' \\
41
63
  --data '{
@@ -52,34 +74,64 @@ export const snippetTextGeneration = (
52
74
  })},
53
75
  "stream": ${!!streaming}
54
76
  }'`,
55
- };
77
+ },
78
+ ];
56
79
  } else {
57
- return snippetBasic(model, accessToken);
80
+ return snippetBasic(model, accessToken, provider);
58
81
  }
59
82
  };
60
83
 
61
- export const snippetZeroShotClassification = (model: ModelDataMinimal, accessToken: string): InferenceSnippet => ({
62
- content: `curl https://api-inference.huggingface.co/models/${model.id} \\
84
+ export const snippetZeroShotClassification = (
85
+ model: ModelDataMinimal,
86
+ accessToken: string,
87
+ provider: InferenceProvider
88
+ ): InferenceSnippet[] => {
89
+ if (provider !== "hf-inference") {
90
+ return [];
91
+ }
92
+ return [
93
+ {
94
+ client: "curl",
95
+ content: `curl https://api-inference.huggingface.co/models/${model.id} \\
63
96
  -X POST \\
64
97
  -d '{"inputs": ${getModelInputSnippet(model, true)}, "parameters": {"candidate_labels": ["refund", "legal", "faq"]}}' \\
65
98
  -H 'Content-Type: application/json' \\
66
99
  -H 'Authorization: Bearer ${accessToken || `{API_TOKEN}`}'`,
67
- });
100
+ },
101
+ ];
102
+ };
68
103
 
69
- export const snippetFile = (model: ModelDataMinimal, accessToken: string): InferenceSnippet => ({
70
- content: `curl https://api-inference.huggingface.co/models/${model.id} \\
104
+ export const snippetFile = (
105
+ model: ModelDataMinimal,
106
+ accessToken: string,
107
+ provider: InferenceProvider
108
+ ): InferenceSnippet[] => {
109
+ if (provider !== "hf-inference") {
110
+ return [];
111
+ }
112
+ return [
113
+ {
114
+ client: "curl",
115
+ content: `curl https://api-inference.huggingface.co/models/${model.id} \\
71
116
  -X POST \\
72
117
  --data-binary '@${getModelInputSnippet(model, true, true)}' \\
73
118
  -H 'Authorization: Bearer ${accessToken || `{API_TOKEN}`}'`,
74
- });
119
+ },
120
+ ];
121
+ };
75
122
 
76
123
  export const curlSnippets: Partial<
77
124
  Record<
78
125
  PipelineType,
79
- (model: ModelDataMinimal, accessToken: string, opts?: Record<string, unknown>) => InferenceSnippet
126
+ (
127
+ model: ModelDataMinimal,
128
+ accessToken: string,
129
+ provider: InferenceProvider,
130
+ opts?: Record<string, unknown>
131
+ ) => InferenceSnippet[]
80
132
  >
81
133
  > = {
82
- // Same order as in js/src/lib/interfaces/Types.ts
134
+ // Same order as in tasks/src/pipelines.ts
83
135
  "text-classification": snippetBasic,
84
136
  "token-classification": snippetBasic,
85
137
  "table-question-answering": snippetBasic,
@@ -108,13 +160,10 @@ export const curlSnippets: Partial<
108
160
  export function getCurlInferenceSnippet(
109
161
  model: ModelDataMinimal,
110
162
  accessToken: string,
163
+ provider: InferenceProvider,
111
164
  opts?: Record<string, unknown>
112
- ): InferenceSnippet {
165
+ ): InferenceSnippet[] {
113
166
  return model.pipeline_tag && model.pipeline_tag in curlSnippets
114
- ? curlSnippets[model.pipeline_tag]?.(model, accessToken, opts) ?? { content: "" }
115
- : { content: "" };
116
- }
117
-
118
- export function hasCurlInferenceSnippet(model: Pick<ModelDataMinimal, "pipeline_tag">): boolean {
119
- return !!model.pipeline_tag && model.pipeline_tag in curlSnippets;
167
+ ? curlSnippets[model.pipeline_tag]?.(model, accessToken, provider, opts) ?? []
168
+ : [];
120
169
  }