@huggingface/tasks 0.10.16 → 0.10.18

package/dist/index.cjs

@@ -4168,6 +4168,24 @@ var asteroid = (model) => [
 
 model = BaseModel.from_pretrained("${model.id}")`
 ];
+var audioseal = (model) => {
+  const watermarkSnippet = `# Watermark Generator
+from audioseal import AudioSeal
+
+model = AudioSeal.load_generator("${model.id}")
+# pass a tensor (tensor_wav) of shape (batch, channels, samples) and a sample rate
+wav, sr = tensor_wav, 16000
+	
+watermark = model.get_watermark(wav, sr)
+watermarked_audio = wav + watermark`;
+  const detectorSnippet = `# Watermark Detector
+from audioseal import AudioSeal
+
+detector = AudioSeal.load_detector("${model.id}")
+	
+result, message = detector.detect_watermark(watermarked_audio, sr)`;
+  return [watermarkSnippet, detectorSnippet];
+};
 function get_base_diffusers_model(model) {
   return model.cardData?.base_model?.toString() ?? "fill-in-base-model";
 }
@@ -4751,9 +4769,7 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     docsUrl: "https://huggingface.co/docs/hub/adapters",
     snippets: adapters,
     filter: true,
-    countDownloads: {
-      term: { path: "adapter_config.json" }
-    }
+    countDownloads: `path:"adapter_config.json"`
   },
   allennlp: {
     prettyLabel: "AllenNLP",
@@ -4770,9 +4786,7 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     docsUrl: "https://huggingface.co/docs/hub/asteroid",
     snippets: asteroid,
     filter: true,
-    countDownloads: {
-      term: { path: "pytorch_model.bin" }
-    }
+    countDownloads: `path:"pytorch_model.bin"`
   },
   audiocraft: {
     prettyLabel: "Audiocraft",
@@ -4781,6 +4795,14 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     snippets: audiocraft,
     filter: false
   },
+  audioseal: {
+    prettyLabel: "AudioSeal",
+    repoName: "audioseal",
+    repoUrl: "https://github.com/facebookresearch/audioseal",
+    filter: false,
+    countDownloads: "path_extension:pt",
+    snippets: audioseal
+  },
   bertopic: {
     prettyLabel: "BERTopic",
     repoName: "BERTopic",
@@ -4793,9 +4815,7 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     repoName: "big_vision",
     repoUrl: "https://github.com/google-research/big_vision",
     filter: false,
-    countDownloads: {
-      wildcard: { path: "*.npz" }
-    }
+    countDownloads: `path_extension:"npz"`
   },
   bm25s: {
     prettyLabel: "BM25S",
@@ -4803,17 +4823,15 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     repoUrl: "https://github.com/xhluca/bm25s",
     snippets: bm25s,
     filter: false,
-    countDownloads: {
-      term: { path: "params.index.json" }
-    }
+    countDownloads: `path:"params.index.json"`
   },
   chat_tts: {
     prettyLabel: "ChatTTS",
     repoName: "ChatTTS",
     repoUrl: "https://github.com/2noise/ChatTTS.git",
+    snippets: chattts,
     filter: false,
-    countDownloads: { term: { path: "asset/GPT.pt" } },
-    snippets: chattts
+    countDownloads: `path:"asset/GPT.pt"`
   },
   diffusers: {
     prettyLabel: "Diffusers",
@@ -4836,18 +4854,14 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     docsUrl: "https://aphp.github.io/edsnlp/latest/",
     filter: false,
     snippets: edsnlp,
-    countDownloads: {
-      wildcard: { path: "*/config.cfg" }
-    }
+    countDownloads: `path_filename:"config" AND path_extension:"cfg"`
   },
   elm: {
     prettyLabel: "ELM",
     repoName: "elm",
     repoUrl: "https://github.com/slicex-ai/elm",
     filter: false,
-    countDownloads: {
-      wildcard: { path: "*/slicex_elm_config.json" }
-    }
+    countDownloads: `path_filename:"slicex_elm_config" AND path_extension:"json"`
   },
   espnet: {
     prettyLabel: "ESPnet",
@@ -4878,9 +4892,7 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     repoUrl: "https://fasttext.cc/",
     snippets: fasttext,
     filter: true,
-    countDownloads: {
-      wildcard: { path: "*.bin" }
-    }
+    countDownloads: `path_extension:"bin"`
   },
   flair: {
     prettyLabel: "Flair",
@@ -4889,16 +4901,14 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     docsUrl: "https://huggingface.co/docs/hub/flair",
     snippets: flair,
     filter: true,
-    countDownloads: {
-      term: { path: "pytorch_model.bin" }
-    }
+    countDownloads: `path:"pytorch_model.bin"`
   },
   "gemma.cpp": {
     prettyLabel: "gemma.cpp",
     repoName: "gemma.cpp",
     repoUrl: "https://github.com/google/gemma.cpp",
     filter: false,
-    countDownloads: { wildcard: { path: "*.sbs" } }
+    countDownloads: `path_extension:"sbs"`
   },
   gliner: {
     prettyLabel: "GLiNER",
@@ -4906,26 +4916,20 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     repoUrl: "https://github.com/urchade/GLiNER",
     snippets: gliner,
     filter: false,
-    countDownloads: {
-      term: { path: "gliner_config.json" }
-    }
+    countDownloads: `path:"gliner_config.json"`
   },
   grok: {
     prettyLabel: "Grok",
     repoName: "Grok",
     repoUrl: "https://github.com/xai-org/grok-1",
     filter: false,
-    countDownloads: {
-      terms: { path: ["ckpt/tensor00000_000", "ckpt-0/tensor00000_000"] }
-    }
+    countDownloads: `path:"ckpt/tensor00000_000" OR path:"ckpt-0/tensor00000_000"`
   },
   "hunyuan-dit": {
     prettyLabel: "HunyuanDiT",
     repoName: "HunyuanDiT",
     repoUrl: "https://github.com/Tencent/HunyuanDiT",
-    countDownloads: {
-      terms: { path: ["pytorch_model_ema.pt", "pytorch_model_distill.pt"] }
-    }
+    countDownloads: `path:"pytorch_model_ema.pt" OR path:"pytorch_model_distill.pt"`
   },
   keras: {
     prettyLabel: "Keras",
@@ -4934,7 +4938,7 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     docsUrl: "https://huggingface.co/docs/hub/keras",
     snippets: keras,
     filter: true,
-    countDownloads: { term: { path: "saved_model.pb" } }
+    countDownloads: `path:"saved_model.pb"`
   },
   "keras-nlp": {
     prettyLabel: "KerasNLP",
@@ -4960,7 +4964,7 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     docsUrl: "https://huggingface.co/docs/hub/ml-agents",
     snippets: mlAgents,
     filter: true,
-    countDownloads: { wildcard: { path: "*.onnx" } }
+    countDownloads: `path_extension:"onnx"`
   },
   mlx: {
     prettyLabel: "MLX",
@@ -4976,7 +4980,7 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     docsUrl: "https://huggingface.co/docs/hub/mlx-image",
     snippets: mlxim,
     filter: false,
-    countDownloads: { term: { path: "model.safetensors" } }
+    countDownloads: `path:"model.safetensors"`
   },
   nemo: {
     prettyLabel: "NeMo",
@@ -4984,7 +4988,7 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     repoUrl: "https://github.com/NVIDIA/NeMo",
     snippets: nemo,
     filter: true,
-    countDownloads: { wildcard: { path: "*.nemo" } }
+    countDownloads: `path_extension:"nemo" OR path:"model_config.yaml"`
   },
   open_clip: {
     prettyLabel: "OpenCLIP",
@@ -4992,7 +4996,7 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     repoUrl: "https://github.com/mlfoundations/open_clip",
     snippets: open_clip,
     filter: true,
-    countDownloads: { wildcard: { path: "*pytorch_model.bin" } }
+    countDownloads: `path_extension:"bin" AND path_filename:*pytorch_model`
   },
   paddlenlp: {
     prettyLabel: "paddlenlp",
@@ -5001,9 +5005,7 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     docsUrl: "https://huggingface.co/docs/hub/paddlenlp",
     snippets: paddlenlp,
     filter: true,
-    countDownloads: {
-      term: { path: "model_config.json" }
-    }
+    countDownloads: `path:"model_config.json"`
   },
   peft: {
     prettyLabel: "PEFT",
@@ -5011,9 +5013,7 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     repoUrl: "https://github.com/huggingface/peft",
     snippets: peft,
     filter: true,
-    countDownloads: {
-      term: { path: "adapter_config.json" }
-    }
+    countDownloads: `path:"adapter_config.json"`
   },
   "pyannote-audio": {
     prettyLabel: "pyannote.audio",
@@ -5034,7 +5034,7 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     repoName: "recurrentgemma",
     repoUrl: "https://github.com/google-deepmind/recurrentgemma",
     filter: false,
-    countDownloads: { term: { path: "tokenizer.model" } }
+    countDownloads: `path:"tokenizer.model"`
   },
   "sample-factory": {
     prettyLabel: "sample-factory",
@@ -5043,7 +5043,7 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     docsUrl: "https://huggingface.co/docs/hub/sample-factory",
     snippets: sampleFactory,
     filter: true,
-    countDownloads: { term: { path: "cfg.json" } }
+    countDownloads: `path:"cfg.json"`
   },
   "sentence-transformers": {
     prettyLabel: "sentence-transformers",
@@ -5067,9 +5067,7 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     repoUrl: "https://github.com/scikit-learn/scikit-learn",
     snippets: sklearn,
     filter: true,
-    countDownloads: {
-      term: { path: "sklearn_model.joblib" }
-    }
+    countDownloads: `path:"sklearn_model.joblib"`
   },
   spacy: {
     prettyLabel: "spaCy",
@@ -5078,9 +5076,7 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     docsUrl: "https://huggingface.co/docs/hub/spacy",
     snippets: spacy,
     filter: true,
-    countDownloads: {
-      wildcard: { path: "*.whl" }
-    }
+    countDownloads: `path_extension:"whl"`
   },
   "span-marker": {
     prettyLabel: "SpanMarker",
@@ -5097,18 +5093,23 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     docsUrl: "https://huggingface.co/docs/hub/speechbrain",
     snippets: speechbrain,
     filter: true,
-    countDownloads: {
-      term: { path: "hyperparams.yaml" }
-    }
+    countDownloads: `path:"hyperparams.yaml"`
   },
   "stable-audio-tools": {
     prettyLabel: "Stable Audio Tools",
     repoName: "stable-audio-tools",
     repoUrl: "https://github.com/Stability-AI/stable-audio-tools.git",
     filter: false,
-    countDownloads: { term: { path: "model.safetensors" } },
+    countDownloads: `path:"model.safetensors"`,
     snippets: stable_audio_tools
   },
+  "diffusion-single-file": {
+    prettyLabel: "Diffusion Single File",
+    repoName: "diffusion-single-file",
+    repoUrl: "https://github.com/comfyanonymous/ComfyUI",
+    filter: false,
+    countDownloads: `path_extension:"safetensors"`
+  },
   "stable-baselines3": {
     prettyLabel: "stable-baselines3",
     repoName: "stable-baselines3",
@@ -5116,9 +5117,7 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     docsUrl: "https://huggingface.co/docs/hub/stable-baselines3",
     snippets: stableBaselines3,
     filter: true,
-    countDownloads: {
-      wildcard: { path: "*.zip" }
-    }
+    countDownloads: `path_extension:"zip"`
   },
   stanza: {
     prettyLabel: "Stanza",
@@ -5127,9 +5126,7 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     docsUrl: "https://huggingface.co/docs/hub/stanza",
     snippets: stanza,
     filter: true,
-    countDownloads: {
-      term: { path: "models/default.zip" }
-    }
+    countDownloads: `path:"models/default.zip"`
   },
   tensorflowtts: {
     prettyLabel: "TensorFlowTTS",
@@ -5142,16 +5139,14 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     repoName: "TiC-CLIP",
     repoUrl: "https://github.com/apple/ml-tic-clip",
     filter: false,
-    countDownloads: { wildcard: { path: "checkpoints/*.pt" } }
+    countDownloads: `path_extension:"pt" AND path_prefix:"checkpoints/"`
   },
   timesfm: {
     prettyLabel: "TimesFM",
     repoName: "timesfm",
     repoUrl: "https://github.com/google-research/timesfm",
     filter: false,
-    countDownloads: {
-      term: { path: "checkpoints/checkpoint_1100000/state/checkpoint" }
-    }
+    countDownloads: `path:"checkpoints/checkpoint_1100000/state/checkpoint"`
   },
   timm: {
     prettyLabel: "timm",
@@ -5160,9 +5155,7 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     docsUrl: "https://huggingface.co/docs/hub/timm",
     snippets: timm,
     filter: true,
-    countDownloads: {
-      terms: { path: ["pytorch_model.bin", "model.safetensors"] }
-    }
+    countDownloads: `path:"pytorch_model.bin" OR path:"model.safetensors"`
   },
   transformers: {
     prettyLabel: "Transformers",
@@ -5186,9 +5179,7 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     repoUrl: "https://github.com/Unity-Technologies/sentis-samples",
     snippets: sentis,
     filter: true,
-    countDownloads: {
-      wildcard: { path: "*.sentis" }
-    }
+    countDownloads: `path_extension:"sentis"`
   },
   voicecraft: {
     prettyLabel: "VoiceCraft",
@@ -5201,9 +5192,7 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     prettyLabel: "WhisperKit",
     repoName: "WhisperKit",
     repoUrl: "https://github.com/argmaxinc/WhisperKit",
-    countDownloads: {
-      wildcard: { path: "*/model.mil" }
-    }
+    countDownloads: `path_filename:"model" AND path_extension:"mil" AND _exists_:"path_prefix"`
   }
 };
 var ALL_MODEL_LIBRARY_KEYS = Object.keys(MODEL_LIBRARIES_UI_ELEMENTS);