@huggingface/tasks 0.2.0 → 0.2.2

package/dist/index.d.ts

@@ -5,7 +5,7 @@
  * File formats live in an enum inside the internal codebase.
  */
 declare enum ModelLibrary {
-    "adapter-transformers" = "Adapter Transformers",
+    "adapter-transformers" = "Adapters",
     "allennlp" = "allenNLP",
     "asteroid" = "Asteroid",
     "bertopic" = "BERTopic",
@@ -37,12 +37,13 @@ declare enum ModelLibrary {
     "stanza" = "Stanza",
     "fasttext" = "fastText",
     "stable-baselines3" = "Stable-Baselines3",
-    "ml-agents" = "ML-Agents",
+    "ml-agents" = "Unity ML-Agents",
     "pythae" = "Pythae",
-    "mindspore" = "MindSpore"
+    "mindspore" = "MindSpore",
+    "unity-sentis" = "Unity Sentis"
 }
 type ModelLibraryKey = keyof typeof ModelLibrary;
-declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("speechbrain" | "fastai" | "adapter-transformers" | "allennlp" | "asteroid" | "bertopic" | "diffusers" | "doctr" | "espnet" | "fairseq" | "flair" | "keras" | "k2" | "mlx" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "tensorflowtts" | "timm" | "transformers" | "transformers.js" | "stanza" | "fasttext" | "stable-baselines3" | "ml-agents" | "pythae" | "mindspore")[];
+declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("speechbrain" | "fastai" | "adapter-transformers" | "allennlp" | "asteroid" | "bertopic" | "diffusers" | "doctr" | "espnet" | "fairseq" | "flair" | "keras" | "k2" | "mlx" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "tensorflowtts" | "timm" | "transformers" | "transformers.js" | "stanza" | "fasttext" | "stable-baselines3" | "ml-agents" | "pythae" | "mindspore" | "unity-sentis")[];
 
 declare const MODALITIES: readonly ["cv", "nlp", "audio", "tabular", "multimodal", "rl", "other"];
 type Modality = (typeof MODALITIES)[number];
@@ -447,7 +448,7 @@ declare const PIPELINE_TYPES_SET: Set<"other" | "text-classification" | "token-c
 
 /**
  * Mapping from library name (excluding Transformers) to its supported tasks.
- * Inference API should be disabled for all other (library, task) pairs beyond this mapping.
+ * Inference Endpoints (serverless) should be disabled for all other (library, task) pairs beyond this mapping.
  * As an exception, we assume Transformers supports all inference tasks.
  * This mapping is generated automatically by "python-api-export-tasks" action in huggingface/api-inference-community repo upon merge.
  * Ref: https://github.com/huggingface/api-inference-community/pull/158
@@ -611,7 +612,7 @@ interface ModelData {
      */
     widgetData?: WidgetExample[] | undefined;
     /**
-     * Parameters that will be used by the widget when calling Inference API
+     * Parameters that will be used by the widget when calling Inference Endpoints (serverless)
      * https://huggingface.co/docs/api-inference/detailed_parameters
      *
      * can be set in the model card metadata (under `inference/parameters`)
@@ -730,6 +731,7 @@ interface TaskData {
     datasets: ExampleRepo[];
     demo: TaskDemo;
     id: PipelineType;
+    canonicalId?: PipelineType;
     isPlaceholder?: boolean;
     label: string;
     libraries: ModelLibraryKey[];