@huggingface/tasks 0.10.21 → 0.10.22

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/dist/index.cjs CHANGED
@@ -4296,13 +4296,18 @@ var gliner = (model) => [
4296
4296
  model = GLiNER.from_pretrained("${model.id}")`
4297
4297
  ];
4298
4298
  var keras = (model) => [
4299
- `from huggingface_hub import from_pretrained_keras
4299
+ `# Available backend options are: "jax", "tensorflow", "torch".
4300
+ import os
4301
+ os.environ["KERAS_BACKEND"] = "tensorflow"
4302
+
4303
+ import keras
4300
4304
 
4301
- model = from_pretrained_keras("${model.id}")
4305
+ model = keras.saving.load_model("hf://${model.id}")
4302
4306
  `
4303
4307
  ];
4304
4308
  var keras_nlp = (model) => [
4305
4309
  `# Available backend options are: "jax", "tensorflow", "torch".
4310
+ import os
4306
4311
  os.environ["KERAS_BACKEND"] = "tensorflow"
4307
4312
 
4308
4313
  import keras_nlp
@@ -4311,6 +4316,20 @@ tokenizer = keras_nlp.models.Tokenizer.from_preset("hf://${model.id}")
4311
4316
  backbone = keras_nlp.models.Backbone.from_preset("hf://${model.id}")
4312
4317
  `
4313
4318
  ];
4319
+ var tf_keras = (model) => [
4320
+ `# Note: 'keras<3.x' or 'tf_keras' must be installed (legacy)
4321
+ # See https://github.com/keras-team/tf-keras for more details.
4322
+ from huggingface_hub import from_pretrained_keras
4323
+
4324
+ model = from_pretrained_keras("${model.id}")
4325
+ `
4326
+ ];
4327
+ var mars5_tts = (model) => [
4328
+ `# Install from https://github.com/Camb-ai/MARS5-TTS
4329
+
4330
+ from inference import Mars5TTS
4331
+ mars5 = Mars5TTS.from_pretrained("${model.id}")`
4332
+ ];
4314
4333
  var open_clip = (model) => [
4315
4334
  `import open_clip
4316
4335
 
@@ -4759,6 +4778,19 @@ var audiocraft = (model) => {
4759
4778
  return [`# Type of model unknown.`];
4760
4779
  }
4761
4780
  };
4781
+ var whisperkit = () => [
4782
+ `# Install CLI with Homebrew on macOS device
4783
+ brew install whisperkit-cli
4784
+
4785
+ # View all available inference options
4786
+ whisperkit-cli transcribe --help
4787
+
4788
+ # Download and run inference using whisper base model
4789
+ whisperkit-cli transcribe --audio-path /path/to/audio.mp3
4790
+
4791
+ # Or use your preferred model variant
4792
+ whisperkit-cli transcribe --model "large-v3" --model-prefix "distil" --audio-path /path/to/audio.mp3 --verbose`
4793
+ ];
4762
4794
 
4763
4795
  // src/model-libraries.ts
4764
4796
  var MODEL_LIBRARIES_UI_ELEMENTS = {
@@ -4793,7 +4825,8 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
4793
4825
  repoName: "audiocraft",
4794
4826
  repoUrl: "https://github.com/facebookresearch/audiocraft",
4795
4827
  snippets: audiocraft,
4796
- filter: false
4828
+ filter: false,
4829
+ countDownloads: `path:"state_dict.bin"`
4797
4830
  },
4798
4831
  audioseal: {
4799
4832
  prettyLabel: "AudioSeal",
@@ -4945,6 +4978,16 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
4945
4978
  docsUrl: "https://huggingface.co/docs/hub/keras",
4946
4979
  snippets: keras,
4947
4980
  filter: true,
4981
+ countDownloads: `path:"config.json" OR path_extension:"keras"`
4982
+ },
4983
+ "tf-keras": {
4984
+ // Legacy "Keras 2" library (tensorflow-only)
4985
+ prettyLabel: "TF-Keras",
4986
+ repoName: "TF-Keras",
4987
+ repoUrl: "https://github.com/keras-team/tf-keras",
4988
+ docsUrl: "https://huggingface.co/docs/hub/tf-keras",
4989
+ snippets: tf_keras,
4990
+ filter: true,
4948
4991
  countDownloads: `path:"saved_model.pb"`
4949
4992
  },
4950
4993
  "keras-nlp": {
@@ -4964,6 +5007,14 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
4964
5007
  repoName: "mindspore",
4965
5008
  repoUrl: "https://github.com/mindspore-ai/mindspore"
4966
5009
  },
5010
+ "mars5-tts": {
5011
+ prettyLabel: "MARS5-TTS",
5012
+ repoName: "MARS5-TTS",
5013
+ repoUrl: "https://github.com/Camb-ai/MARS5-TTS",
5014
+ filter: false,
5015
+ countDownloads: `path:"mars5_ar.safetensors"`,
5016
+ snippets: mars5_tts
5017
+ },
4967
5018
  "ml-agents": {
4968
5019
  prettyLabel: "ml-agents",
4969
5020
  repoName: "ml-agents",
@@ -5199,6 +5250,8 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
5199
5250
  prettyLabel: "WhisperKit",
5200
5251
  repoName: "WhisperKit",
5201
5252
  repoUrl: "https://github.com/argmaxinc/WhisperKit",
5253
+ docsUrl: "https://github.com/argmaxinc/WhisperKit?tab=readme-ov-file#homebrew",
5254
+ snippets: whisperkit,
5202
5255
  countDownloads: `path_filename:"model" AND path_extension:"mil" AND _exists_:"path_prefix"`
5203
5256
  }
5204
5257
  };
package/dist/index.js CHANGED
@@ -4257,13 +4257,18 @@ var gliner = (model) => [
4257
4257
  model = GLiNER.from_pretrained("${model.id}")`
4258
4258
  ];
4259
4259
  var keras = (model) => [
4260
- `from huggingface_hub import from_pretrained_keras
4260
+ `# Available backend options are: "jax", "tensorflow", "torch".
4261
+ import os
4262
+ os.environ["KERAS_BACKEND"] = "tensorflow"
4263
+
4264
+ import keras
4261
4265
 
4262
- model = from_pretrained_keras("${model.id}")
4266
+ model = keras.saving.load_model("hf://${model.id}")
4263
4267
  `
4264
4268
  ];
4265
4269
  var keras_nlp = (model) => [
4266
4270
  `# Available backend options are: "jax", "tensorflow", "torch".
4271
+ import os
4267
4272
  os.environ["KERAS_BACKEND"] = "tensorflow"
4268
4273
 
4269
4274
  import keras_nlp
@@ -4272,6 +4277,20 @@ tokenizer = keras_nlp.models.Tokenizer.from_preset("hf://${model.id}")
4272
4277
  backbone = keras_nlp.models.Backbone.from_preset("hf://${model.id}")
4273
4278
  `
4274
4279
  ];
4280
+ var tf_keras = (model) => [
4281
+ `# Note: 'keras<3.x' or 'tf_keras' must be installed (legacy)
4282
+ # See https://github.com/keras-team/tf-keras for more details.
4283
+ from huggingface_hub import from_pretrained_keras
4284
+
4285
+ model = from_pretrained_keras("${model.id}")
4286
+ `
4287
+ ];
4288
+ var mars5_tts = (model) => [
4289
+ `# Install from https://github.com/Camb-ai/MARS5-TTS
4290
+
4291
+ from inference import Mars5TTS
4292
+ mars5 = Mars5TTS.from_pretrained("${model.id}")`
4293
+ ];
4275
4294
  var open_clip = (model) => [
4276
4295
  `import open_clip
4277
4296
 
@@ -4720,6 +4739,19 @@ var audiocraft = (model) => {
4720
4739
  return [`# Type of model unknown.`];
4721
4740
  }
4722
4741
  };
4742
+ var whisperkit = () => [
4743
+ `# Install CLI with Homebrew on macOS device
4744
+ brew install whisperkit-cli
4745
+
4746
+ # View all available inference options
4747
+ whisperkit-cli transcribe --help
4748
+
4749
+ # Download and run inference using whisper base model
4750
+ whisperkit-cli transcribe --audio-path /path/to/audio.mp3
4751
+
4752
+ # Or use your preferred model variant
4753
+ whisperkit-cli transcribe --model "large-v3" --model-prefix "distil" --audio-path /path/to/audio.mp3 --verbose`
4754
+ ];
4723
4755
 
4724
4756
  // src/model-libraries.ts
4725
4757
  var MODEL_LIBRARIES_UI_ELEMENTS = {
@@ -4754,7 +4786,8 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
4754
4786
  repoName: "audiocraft",
4755
4787
  repoUrl: "https://github.com/facebookresearch/audiocraft",
4756
4788
  snippets: audiocraft,
4757
- filter: false
4789
+ filter: false,
4790
+ countDownloads: `path:"state_dict.bin"`
4758
4791
  },
4759
4792
  audioseal: {
4760
4793
  prettyLabel: "AudioSeal",
@@ -4906,6 +4939,16 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
4906
4939
  docsUrl: "https://huggingface.co/docs/hub/keras",
4907
4940
  snippets: keras,
4908
4941
  filter: true,
4942
+ countDownloads: `path:"config.json" OR path_extension:"keras"`
4943
+ },
4944
+ "tf-keras": {
4945
+ // Legacy "Keras 2" library (tensorflow-only)
4946
+ prettyLabel: "TF-Keras",
4947
+ repoName: "TF-Keras",
4948
+ repoUrl: "https://github.com/keras-team/tf-keras",
4949
+ docsUrl: "https://huggingface.co/docs/hub/tf-keras",
4950
+ snippets: tf_keras,
4951
+ filter: true,
4909
4952
  countDownloads: `path:"saved_model.pb"`
4910
4953
  },
4911
4954
  "keras-nlp": {
@@ -4925,6 +4968,14 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
4925
4968
  repoName: "mindspore",
4926
4969
  repoUrl: "https://github.com/mindspore-ai/mindspore"
4927
4970
  },
4971
+ "mars5-tts": {
4972
+ prettyLabel: "MARS5-TTS",
4973
+ repoName: "MARS5-TTS",
4974
+ repoUrl: "https://github.com/Camb-ai/MARS5-TTS",
4975
+ filter: false,
4976
+ countDownloads: `path:"mars5_ar.safetensors"`,
4977
+ snippets: mars5_tts
4978
+ },
4928
4979
  "ml-agents": {
4929
4980
  prettyLabel: "ml-agents",
4930
4981
  repoName: "ml-agents",
@@ -5160,6 +5211,8 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
5160
5211
  prettyLabel: "WhisperKit",
5161
5212
  repoName: "WhisperKit",
5162
5213
  repoUrl: "https://github.com/argmaxinc/WhisperKit",
5214
+ docsUrl: "https://github.com/argmaxinc/WhisperKit?tab=readme-ov-file#homebrew",
5215
+ snippets: whisperkit,
5163
5216
  countDownloads: `path_filename:"model" AND path_extension:"mil" AND _exists_:"path_prefix"`
5164
5217
  }
5165
5218
  };
@@ -15,6 +15,8 @@ export declare const flair: (model: ModelData) => string[];
15
15
  export declare const gliner: (model: ModelData) => string[];
16
16
  export declare const keras: (model: ModelData) => string[];
17
17
  export declare const keras_nlp: (model: ModelData) => string[];
18
+ export declare const tf_keras: (model: ModelData) => string[];
19
+ export declare const mars5_tts: (model: ModelData) => string[];
18
20
  export declare const open_clip: (model: ModelData) => string[];
19
21
  export declare const paddlenlp: (model: ModelData) => string[];
20
22
  export declare const pyannote_audio_pipeline: (model: ModelData) => string[];
@@ -45,4 +47,5 @@ export declare const mlxim: (model: ModelData) => string[];
45
47
  export declare const nemo: (model: ModelData) => string[];
46
48
  export declare const pythae: (model: ModelData) => string[];
47
49
  export declare const audiocraft: (model: ModelData) => string[];
50
+ export declare const whisperkit: () => string[];
48
51
  //# sourceMappingURL=model-libraries-snippets.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"model-libraries-snippets.d.ts","sourceRoot":"","sources":["../../src/model-libraries-snippets.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAY9C,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAkBF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAkBlD,CAAC;AAMF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AA+BF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAUlD,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAgB/C,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAMlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EASlD,CAAC;AAIF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAO/C,CAAC;AAEF,eAAO,MAAM,OAAO,UAAW,SAAS,KAAG,MAAM,EAMhD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAK9C,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EASlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAKlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAsBlD,CAAC;AAEF,eAAO,MAAM,uBAAuB,UAAW,SAAS,KAAG,MAAM,EAehE,CAAC;AAiBF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAKvD,CAAC;AAyBF,eAAO,MAAM,aAAa,UAAW,SAAS,KAAG,MAAM,EAOtD,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAI7C,CAAC;AAsCF,eAAO,MAAM,OAAO,UAAW,SAAS,KAAG,MAAM,EAehD,CAAC;AAEF,eAAO,MAAM,kBAAkB,UAAW,SAAS,KAAG,MAAM,EAmC3D,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,aAAa,UAAW,SAAS,KAAG,MAAM,EAEtD,CAAC;AAEF,eAAO,MAAM,oBAAoB,UAAW,SAAS,KAAG,MAAM,EAI7D,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAU9C,CAAC;AAEF,eAAO,MAAM,WAAW,UAAW,SAAS,KAAG,MAAM,EAIpD,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAK/C,CAAC;AAkBF,eAAO,MAAM,WAAW,UAAW,SAAS,KAAG,MAAM,EAkBpD,CAAC;AAEF,eAAO,MAAM,YAAY,UAAW,SAAS,KAAG,MAAM,EA4CrD,CAAC;AAEF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAcvD,CAAC;AAiBF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAkB7C,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAEF,eAAO,MAAM,gBAAgB,UAAW,SAAS,KAAG,MAAM,EAMzD,CAAC;AAgBF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAEjD,CAAC;AAEF,eAAO,MAAM,MAAM,QAA6B,MAAM,EAMrD,CAAC;AAEF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAInD,CAAC;AAEF,eAAO,MAAM,OAAO,QAAO,MAAM,EAYhC,CAAC;AAEF,eAAO,MAAM,GAAG,UAAW,SAAS,KAAG,MAAM,EAK5C,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAQ7C,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AA6BF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAUnD,CAAC"}
1
+ {"version":3,"file":"model-libraries-snippets.d.ts","sourceRoot":"","sources":["../../src/model-libraries-snippets.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAY9C,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAkBF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAkBlD,CAAC;AAMF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AA+BF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAUlD,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAgB/C,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAMlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EASlD,CAAC;AAIF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAO/C,CAAC;AAEF,eAAO,MAAM,OAAO,UAAW,SAAS,KAAG,MAAM,EAMhD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAS9C,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAUlD,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAOjD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAKlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAKlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAsBlD,CAAC;AAEF,eAAO,MAAM,uBAAuB,UAAW,SAAS,KAAG,MAAM,EAehE,CAAC;AAiBF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAKvD,CAAC;AAyBF,eAAO,MAAM,aAAa,UAAW,SAAS,KAAG,MAAM,EAOtD,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAI7C,CAAC;AAsCF,eAAO,MAAM,OAAO,UAAW,SAAS,KAAG,MAAM,EAehD,CAAC;AAEF,eAAO,MAAM,kBAAkB,UAAW,SAAS,KAAG,MAAM,EAmC3D,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,aAAa,UAAW,SAAS,KAAG,MAAM,EAEtD,CAAC;AAEF,eAAO,MAAM,oBAAoB,UAAW,SAAS,KAAG,MAAM,EAI7D,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAU9C,CAAC;AAEF,eAAO,MAAM,WAAW,UAAW,SAAS,KAAG,MAAM,EAIpD,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAK/C,CAAC;AAkBF,eAAO,MAAM,WAAW,UAAW,SAAS,KAAG,MAAM,EAkBpD,CAAC;AAEF,eAAO,MAAM,YAAY,UAAW,SAAS,KAAG,MAAM,EA4CrD,CAAC;AAEF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAcvD,CAAC;AAiBF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAkB7C,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAEF,eAAO,MAAM,gBAAgB,UAAW,SAAS,KAAG,MAAM,EAMzD,CAAC;AAgBF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAEjD,CAAC;AAEF,eAAO,MAAM,MAAM,QAA6B,MAAM,EAMrD,CAAC;AAEF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAInD,CAAC;AAEF,eAAO,MAAM,OAAO,QAAO,MAAM,EAYhC,CAAC;AAEF,eAAO,MAAM,GAAG,UAAW,SAAS,KAAG,MAAM,EAK5C,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAQ7C,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AA6BF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAUnD,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAYnC,CAAC"}
@@ -86,6 +86,7 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
86
86
  repoUrl: string;
87
87
  snippets: (model: ModelData) => string[];
88
88
  filter: false;
89
+ countDownloads: string;
89
90
  };
90
91
  audioseal: {
91
92
  prettyLabel: string;
@@ -238,6 +239,15 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
238
239
  filter: true;
239
240
  countDownloads: string;
240
241
  };
242
+ "tf-keras": {
243
+ prettyLabel: string;
244
+ repoName: string;
245
+ repoUrl: string;
246
+ docsUrl: string;
247
+ snippets: (model: ModelData) => string[];
248
+ filter: true;
249
+ countDownloads: string;
250
+ };
241
251
  "keras-nlp": {
242
252
  prettyLabel: string;
243
253
  repoName: string;
@@ -255,6 +265,14 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
255
265
  repoName: string;
256
266
  repoUrl: string;
257
267
  };
268
+ "mars5-tts": {
269
+ prettyLabel: string;
270
+ repoName: string;
271
+ repoUrl: string;
272
+ filter: false;
273
+ countDownloads: string;
274
+ snippets: (model: ModelData) => string[];
275
+ };
258
276
  "ml-agents": {
259
277
  prettyLabel: string;
260
278
  repoName: string;
@@ -490,10 +508,12 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
490
508
  prettyLabel: string;
491
509
  repoName: string;
492
510
  repoUrl: string;
511
+ docsUrl: string;
512
+ snippets: () => string[];
493
513
  countDownloads: string;
494
514
  };
495
515
  };
496
516
  export type ModelLibraryKey = keyof typeof MODEL_LIBRARIES_UI_ELEMENTS;
497
- export declare const ALL_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "audioseal" | "bertopic" | "big_vision" | "bm25s" | "chat_tts" | "diffusers" | "doctr" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gemma.cpp" | "gliner" | "glyph-byt5" | "grok" | "hunyuan-dit" | "keras" | "keras-nlp" | "k2" | "mindspore" | "ml-agents" | "mlx" | "mlx-image" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "recurrentgemma" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "stable-audio-tools" | "diffusion-single-file" | "stable-baselines3" | "stanza" | "tensorflowtts" | "tic-clip" | "timesfm" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "voicecraft" | "whisperkit")[];
498
- export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "audioseal" | "bertopic" | "big_vision" | "bm25s" | "chat_tts" | "diffusers" | "doctr" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gemma.cpp" | "gliner" | "glyph-byt5" | "grok" | "hunyuan-dit" | "keras" | "keras-nlp" | "k2" | "mindspore" | "ml-agents" | "mlx" | "mlx-image" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "recurrentgemma" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "stable-audio-tools" | "diffusion-single-file" | "stable-baselines3" | "stanza" | "tensorflowtts" | "tic-clip" | "timesfm" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "voicecraft" | "whisperkit")[];
517
+ export declare const ALL_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "audioseal" | "bertopic" | "big_vision" | "bm25s" | "chat_tts" | "diffusers" | "doctr" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gemma.cpp" | "gliner" | "glyph-byt5" | "grok" | "hunyuan-dit" | "keras" | "tf-keras" | "keras-nlp" | "k2" | "mindspore" | "mars5-tts" | "ml-agents" | "mlx" | "mlx-image" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "recurrentgemma" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "stable-audio-tools" | "diffusion-single-file" | "stable-baselines3" | "stanza" | "tensorflowtts" | "tic-clip" | "timesfm" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "voicecraft" | "whisperkit")[];
518
+ export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "audioseal" | "bertopic" | "big_vision" | "bm25s" | "chat_tts" | "diffusers" | "doctr" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gemma.cpp" | "gliner" | "glyph-byt5" | "grok" | "hunyuan-dit" | "keras" | "tf-keras" | "keras-nlp" | "k2" | "mindspore" | "mars5-tts" | "ml-agents" | "mlx" | "mlx-image" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "recurrentgemma" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "stable-audio-tools" | "diffusion-single-file" | "stable-baselines3" | "stanza" | "tensorflowtts" | "tic-clip" | "timesfm" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "voicecraft" | "whisperkit")[];
499
519
  //# sourceMappingURL=model-libraries.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"model-libraries.d.ts","sourceRoot":"","sources":["../../src/model-libraries.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,6BAA6B,CAAC;AAEtE;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;;;OAIG;IACH,WAAW,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,MAAM,EAAE,CAAC;IAC1C;;;;;OAKG;IACH,cAAc,CAAC,EAAE,kBAAkB,CAAC;IACpC;;;OAGG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;CACjB;AAED;;;;;;;;;;;;;GAaG;AAEH,eAAO,MAAM,2BAA2B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAwbI,CAAC;AAE7C,MAAM,MAAM,eAAe,GAAG,MAAM,OAAO,2BAA2B,CAAC;AAEvE,eAAO,MAAM,sBAAsB,myBAAgE,CAAC;AAEpG,eAAO,MAAM,8BAA8B,myBAQ1B,CAAC"}
1
+ {"version":3,"file":"model-libraries.d.ts","sourceRoot":"","sources":["../../src/model-libraries.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,6BAA6B,CAAC;AAEtE;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;;;OAIG;IACH,WAAW,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,MAAM,EAAE,CAAC;IAC1C;;;;;OAKG;IACH,cAAc,CAAC,EAAE,kBAAkB,CAAC;IACpC;;;OAGG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;CACjB;AAED;;;;;;;;;;;;;GAaG;AAEH,eAAO,MAAM,2BAA2B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA6cI,CAAC;AAE7C,MAAM,MAAM,eAAe,GAAG,MAAM,OAAO,2BAA2B,CAAC;AAEvE,eAAO,MAAM,sBAAsB,8zBAAgE,CAAC;AAEpG,eAAO,MAAM,8BAA8B,8zBAQ1B,CAAC"}
package/package.json CHANGED
@@ -1,7 +1,7 @@
1
1
  {
2
2
  "name": "@huggingface/tasks",
3
3
  "packageManager": "pnpm@8.10.5",
4
- "version": "0.10.21",
4
+ "version": "0.10.22",
5
5
  "description": "List of ML tasks for huggingface.co/tasks",
6
6
  "repository": "https://github.com/huggingface/huggingface.js.git",
7
7
  "publishConfig": {
@@ -192,14 +192,19 @@ model = GLiNER.from_pretrained("${model.id}")`,
192
192
  ];
193
193
 
194
194
  export const keras = (model: ModelData): string[] => [
195
- `from huggingface_hub import from_pretrained_keras
195
+ `# Available backend options are: "jax", "tensorflow", "torch".
196
+ import os
197
+ os.environ["KERAS_BACKEND"] = "tensorflow"
198
+
199
+ import keras
196
200
 
197
- model = from_pretrained_keras("${model.id}")
201
+ model = keras.saving.load_model("hf://${model.id}")
198
202
  `,
199
203
  ];
200
204
 
201
205
  export const keras_nlp = (model: ModelData): string[] => [
202
206
  `# Available backend options are: "jax", "tensorflow", "torch".
207
+ import os
203
208
  os.environ["KERAS_BACKEND"] = "tensorflow"
204
209
 
205
210
  import keras_nlp
@@ -209,6 +214,22 @@ backbone = keras_nlp.models.Backbone.from_preset("hf://${model.id}")
209
214
  `,
210
215
  ];
211
216
 
217
+ export const tf_keras = (model: ModelData): string[] => [
218
+ `# Note: 'keras<3.x' or 'tf_keras' must be installed (legacy)
219
+ # See https://github.com/keras-team/tf-keras for more details.
220
+ from huggingface_hub import from_pretrained_keras
221
+
222
+ model = from_pretrained_keras("${model.id}")
223
+ `,
224
+ ];
225
+
226
+ export const mars5_tts = (model: ModelData): string[] => [
227
+ `# Install from https://github.com/Camb-ai/MARS5-TTS
228
+
229
+ from inference import Mars5TTS
230
+ mars5 = Mars5TTS.from_pretrained("${model.id}")`,
231
+ ];
232
+
212
233
  export const open_clip = (model: ModelData): string[] => [
213
234
  `import open_clip
214
235
 
@@ -715,4 +736,18 @@ export const audiocraft = (model: ModelData): string[] => {
715
736
  return [`# Type of model unknown.`];
716
737
  }
717
738
  };
739
+
740
+ export const whisperkit = (): string[] => [
741
+ `# Install CLI with Homebrew on macOS device
742
+ brew install whisperkit-cli
743
+
744
+ # View all available inference options
745
+ whisperkit-cli transcribe --help
746
+
747
+ # Download and run inference using whisper base model
748
+ whisperkit-cli transcribe --audio-path /path/to/audio.mp3
749
+
750
+ # Or use your preferred model variant
751
+ whisperkit-cli transcribe --model "large-v3" --model-prefix "distil" --audio-path /path/to/audio.mp3 --verbose`,
752
+ ];
718
753
  //#endregion
@@ -90,6 +90,7 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
90
90
  repoUrl: "https://github.com/facebookresearch/audiocraft",
91
91
  snippets: snippets.audiocraft,
92
92
  filter: false,
93
+ countDownloads: `path:"state_dict.bin"`,
93
94
  },
94
95
  audioseal: {
95
96
  prettyLabel: "AudioSeal",
@@ -241,6 +242,16 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
241
242
  docsUrl: "https://huggingface.co/docs/hub/keras",
242
243
  snippets: snippets.keras,
243
244
  filter: true,
245
+ countDownloads: `path:"config.json" OR path_extension:"keras"`,
246
+ },
247
+ "tf-keras": {
248
+ // Legacy "Keras 2" library (tensorflow-only)
249
+ prettyLabel: "TF-Keras",
250
+ repoName: "TF-Keras",
251
+ repoUrl: "https://github.com/keras-team/tf-keras",
252
+ docsUrl: "https://huggingface.co/docs/hub/tf-keras",
253
+ snippets: snippets.tf_keras,
254
+ filter: true,
244
255
  countDownloads: `path:"saved_model.pb"`,
245
256
  },
246
257
  "keras-nlp": {
@@ -260,6 +271,14 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
260
271
  repoName: "mindspore",
261
272
  repoUrl: "https://github.com/mindspore-ai/mindspore",
262
273
  },
274
+ "mars5-tts": {
275
+ prettyLabel: "MARS5-TTS",
276
+ repoName: "MARS5-TTS",
277
+ repoUrl: "https://github.com/Camb-ai/MARS5-TTS",
278
+ filter: false,
279
+ countDownloads: `path:"mars5_ar.safetensors"`,
280
+ snippets: snippets.mars5_tts,
281
+ },
263
282
  "ml-agents": {
264
283
  prettyLabel: "ml-agents",
265
284
  repoName: "ml-agents",
@@ -495,6 +514,8 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
495
514
  prettyLabel: "WhisperKit",
496
515
  repoName: "WhisperKit",
497
516
  repoUrl: "https://github.com/argmaxinc/WhisperKit",
517
+ docsUrl: "https://github.com/argmaxinc/WhisperKit?tab=readme-ov-file#homebrew",
518
+ snippets: snippets.whisperkit,
498
519
  countDownloads: `path_filename:"model" AND path_extension:"mil" AND _exists_:"path_prefix"`,
499
520
  },
500
521
  } satisfies Record<string, LibraryUiElement>;