@huggingface/tasks 0.5.2 → 0.6.1

package/dist/index.cjs

@@ -4083,6 +4083,11 @@ var flair = (model) => [
 
 tagger = SequenceTagger.load("${model.id}")`
 ];
+var gliner = (model) => [
+  `from gliner import GLiNER
+
+model = GLiNER.from_pretrained("${model.id}")`
+];
 var keras = (model) => [
   `from huggingface_hub import from_pretrained_keras
 
@@ -4429,6 +4434,11 @@ var mlx = (model) => [
 export HF_HUB_ENABLE_HF_TRANS: string[]FER=1
 huggingface-cli download --local-dir ${nameWithoutNamespace(model.id)} ${model.id}`
 ];
+var mlxim = (model) => [
+  `from mlxim.model import create_model
+
+model = create_model(${model.id})`
+];
 var nemo = (model) => {
   let command = void 0;
   if (model.tags?.includes("automatic-speech-recognition")) {
@@ -4578,6 +4588,25 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
       term: { path: "pytorch_model.bin" }
     }
   },
+  gliner: {
+    prettyLabel: "GLiNER",
+    repoName: "GLiNER",
+    repoUrl: "https://github.com/urchade/GLiNER",
+    snippets: gliner,
+    filter: false,
+    countDownloads: {
+      term: { 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"] }
+    }
+  },
   keras: {
     prettyLabel: "Keras",
     repoName: "Keras",
@@ -4613,6 +4642,15 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     snippets: mlx,
     filter: true
   },
+  "mlx-image": {
+    prettyLabel: "mlx-image",
+    repoName: "mlx-image",
+    repoUrl: "https://github.com/riccardomusmeci/mlx-image",
+    docsUrl: "https://huggingface.co/docs/hub/mlx-image",
+    snippets: mlxim,
+    filter: false,
+    countDownloads: { term: { path: "model.safetensors" } }
+  },
   nemo: {
     prettyLabel: "NeMo",
     repoName: "NeMo",
@@ -4793,6 +4831,14 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     countDownloads: {
       wildcard: { path: "*.sentis" }
     }
+  },
+  whisperkit: {
+    prettyLabel: "WhisperKit",
+    repoName: "WhisperKit",
+    repoUrl: "https://github.com/argmaxinc/WhisperKit",
+    countDownloads: {
+      wildcard: { path: "*/model.mil" }
+    }
   }
 };
 var ALL_MODEL_LIBRARY_KEYS = Object.keys(MODEL_LIBRARIES_UI_ELEMENTS);

package/dist/index.d.ts

@@ -839,6 +839,29 @@ declare const MODEL_LIBRARIES_UI_ELEMENTS: {
             };
         };
     };
+    gliner: {
+        prettyLabel: string;
+        repoName: string;
+        repoUrl: string;
+        snippets: (model: ModelData) => string[];
+        filter: false;
+        countDownloads: {
+            term: {
+                path: string;
+            };
+        };
+    };
+    grok: {
+        prettyLabel: string;
+        repoName: string;
+        repoUrl: string;
+        filter: false;
+        countDownloads: {
+            terms: {
+                path: string[];
+            };
+        };
+    };
     keras: {
         prettyLabel: string;
         repoName: string;
@@ -882,6 +905,19 @@ declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         snippets: (model: ModelData) => string[];
         filter: true;
     };
+    "mlx-image": {
+        prettyLabel: string;
+        repoName: string;
+        repoUrl: string;
+        docsUrl: string;
+        snippets: (model: ModelData) => string[];
+        filter: false;
+        countDownloads: {
+            term: {
+                path: string;
+            };
+        };
+    };
     nemo: {
         prettyLabel: string;
         repoName: string;
@@ -1093,10 +1129,20 @@ declare const MODEL_LIBRARIES_UI_ELEMENTS: {
             };
         };
     };
+    whisperkit: {
+        prettyLabel: string;
+        repoName: string;
+        repoUrl: string;
+        countDownloads: {
+            wildcard: {
+                path: string;
+            };
+        };
+    };
 };
 type ModelLibraryKey = keyof typeof MODEL_LIBRARIES_UI_ELEMENTS;
-declare const ALL_MODEL_LIBRARY_KEYS: ("sklearn" | "adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "bertopic" | "diffusers" | "doctr" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "keras" | "k2" | "mindspore" | "ml-agents" | "mlx" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "sample-factory" | "sentence-transformers" | "setfit" | "spacy" | "span-marker" | "speechbrain" | "stable-baselines3" | "stanza" | "tensorflowtts" | "timm" | "transformers" | "transformers.js" | "unity-sentis")[];
-declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("sklearn" | "adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "bertopic" | "diffusers" | "doctr" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "keras" | "k2" | "mindspore" | "ml-agents" | "mlx" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "sample-factory" | "sentence-transformers" | "setfit" | "spacy" | "span-marker" | "speechbrain" | "stable-baselines3" | "stanza" | "tensorflowtts" | "timm" | "transformers" | "transformers.js" | "unity-sentis")[];
+declare const ALL_MODEL_LIBRARY_KEYS: ("sklearn" | "adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "bertopic" | "diffusers" | "doctr" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gliner" | "grok" | "keras" | "k2" | "mindspore" | "ml-agents" | "mlx" | "mlx-image" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "sample-factory" | "sentence-transformers" | "setfit" | "spacy" | "span-marker" | "speechbrain" | "stable-baselines3" | "stanza" | "tensorflowtts" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "whisperkit")[];
+declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("sklearn" | "adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "bertopic" | "diffusers" | "doctr" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gliner" | "grok" | "keras" | "k2" | "mindspore" | "ml-agents" | "mlx" | "mlx-image" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "sample-factory" | "sentence-transformers" | "setfit" | "spacy" | "span-marker" | "speechbrain" | "stable-baselines3" | "stanza" | "tensorflowtts" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "whisperkit")[];
 
 /**
  * Mapping from library name (excluding Transformers) to its supported tasks.
@@ -1316,6 +1362,154 @@ interface AutomaticSpeechRecognitionOutputChunk {
     [property: string]: unknown;
 }
 
+/**
+ * Inference code generated from the JSON schema spec in ./spec
+ *
+ * Using src/scripts/inference-codegen
+ */
+/**
+ * Inputs for ChatCompletion inference
+ */
+interface ChatCompletionInput {
+    /**
+     * Number between -2.0 and 2.0. Positive values penalize new tokens based on their existing
+     * frequency in the text so far, decreasing the model's likelihood to repeat the same line
+     * verbatim.
+     */
+    frequency_penalty?: number;
+    /**
+     * The maximum number of tokens that can be generated in the chat completion.
+     */
+    max_tokens?: number;
+    messages: ChatCompletionInputMessage[];
+    /**
+     * The random sampling seed.
+     */
+    seed?: number;
+    /**
+     * Stop generating tokens if a stop token is generated.
+     */
+    stop?: ChatCompletionInputStopReason;
+    /**
+     * If set, partial message deltas will be sent.
+     */
+    stream?: boolean;
+    /**
+     * The value used to modulate the logits distribution.
+     */
+    temperature?: number;
+    /**
+     * If set to < 1, only the smallest set of most probable tokens with probabilities that add
+     * up to `top_p` or higher are kept for generation.
+     */
+    top_p?: number;
+    [property: string]: unknown;
+}
+interface ChatCompletionInputMessage {
+    /**
+     * The content of the message.
+     */
+    content: string;
+    role: ChatCompletionMessageRole;
+    [property: string]: unknown;
+}
+/**
+ * The role of the message author.
+ */
+type ChatCompletionMessageRole = "assistant" | "system" | "user";
+/**
+ * Stop generating tokens if a stop token is generated.
+ */
+type ChatCompletionInputStopReason = string[] | string;
+/**
+ * Outputs for Chat Completion inference
+ */
+interface ChatCompletionOutput {
+    /**
+     * A list of chat completion choices.
+     */
+    choices: ChatCompletionOutputChoice[];
+    /**
+     * The Unix timestamp (in seconds) of when the chat completion was created.
+     */
+    created: number;
+    [property: string]: unknown;
+}
+interface ChatCompletionOutputChoice {
+    /**
+     * The reason why the generation was stopped.
+     */
+    finish_reason: ChatCompletionFinishReason;
+    /**
+     * The index of the choice in the list of choices.
+     */
+    index: number;
+    message: ChatCompletionOutputChoiceMessage;
+    [property: string]: unknown;
+}
+/**
+ * The reason why the generation was stopped.
+ *
+ * The generated sequence reached the maximum allowed length
+ *
+ * The model generated an end-of-sentence (EOS) token
+ *
+ * One of the sequence in stop_sequences was generated
+ */
+type ChatCompletionFinishReason = "length" | "eos_token" | "stop_sequence";
+interface ChatCompletionOutputChoiceMessage {
+    /**
+     * The content of the chat completion message.
+     */
+    content: string;
+    role: ChatCompletionMessageRole;
+    [property: string]: unknown;
+}
+/**
+ * Chat Completion Stream Output
+ */
+interface ChatCompletionStreamOutput {
+    /**
+     * A list of chat completion choices.
+     */
+    choices: ChatCompletionStreamOutputChoice[];
+    /**
+     * The Unix timestamp (in seconds) of when the chat completion was created. Each chunk has
+     * the same timestamp.
+     */
+    created: number;
+    [property: string]: unknown;
+}
+interface ChatCompletionStreamOutputChoice {
+    /**
+     * A chat completion delta generated by streamed model responses.
+     */
+    delta: ChatCompletionStreamOutputDelta;
+    /**
+     * The reason why the generation was stopped.
+     */
+    finish_reason?: ChatCompletionFinishReason;
+    /**
+     * The index of the choice in the list of choices.
+     */
+    index: number;
+    [property: string]: unknown;
+}
+/**
+ * A chat completion delta generated by streamed model responses.
+ */
+interface ChatCompletionStreamOutputDelta {
+    /**
+     * The contents of the chunk message.
+     */
+    content?: string;
+    /**
+     * The role of the author of this message.
+     */
+    role?: string;
+    [property: string]: unknown;
+}
+
 /**
  * Inference code generated from the JSON schema spec in ./spec
  *
@@ -2600,6 +2794,10 @@ interface TextGenerationInput {
      * Additional inference parameters
      */
     parameters?: TextGenerationParameters;
+    /**
+     * Whether to stream output tokens
+     */
+    stream?: boolean;
     [property: string]: unknown;
 }
 /**
@@ -2695,16 +2893,16 @@ interface TextGenerationOutputDetails {
     /**
      * Details about additional sequences when best_of is provided
      */
-    best_of_sequences?: TextGenerationSequenceDetails[];
+    best_of_sequences?: TextGenerationOutputSequenceDetails[];
     /**
      * The reason why the generation was stopped.
      */
-    finish_reason: FinishReason;
+    finish_reason: TextGenerationFinishReason;
     /**
      * The number of generated tokens
      */
     generated_tokens: number;
-    prefill: PrefillToken[];
+    prefill: TextGenerationPrefillToken[];
     /**
      * The random seed used for generation
      */
@@ -2712,23 +2910,24 @@ interface TextGenerationOutputDetails {
     /**
      * The generated tokens and associated details
      */
-    tokens: Token[];
-    [property: string]: unknown;
-}
-interface TextGenerationSequenceDetails {
+    tokens: TextGenerationOutputToken[];
     /**
-     * The reason why the generation was stopped.
+     * Most likely tokens
      */
-    finish_reason: FinishReason;
+    top_tokens?: Array<TextGenerationOutputToken[]>;
+    [property: string]: unknown;
+}
+interface TextGenerationOutputSequenceDetails {
+    finish_reason: TextGenerationFinishReason;
     /**
      * The generated text
      */
-    generated_text: number;
+    generated_text: string;
     /**
      * The number of generated tokens
      */
     generated_tokens: number;
-    prefill: PrefillToken[];
+    prefill: TextGenerationPrefillToken[];
     /**
      * The random seed used for generation
      */
@@ -2736,18 +2935,24 @@ interface TextGenerationSequenceDetails {
     /**
      * The generated tokens and associated details
      */
-    tokens: Token[];
+    tokens: TextGenerationOutputToken[];
+    /**
+     * Most likely tokens
+     */
+    top_tokens?: Array<TextGenerationOutputToken[]>;
     [property: string]: unknown;
 }
 /**
- * The generated sequence reached the maximum allowed length
+ * The reason why the generation was stopped.
  *
- * The model generated an end-of-sentence (EOS) token
+ * length: The generated sequence reached the maximum allowed length
  *
- * One of the sequence in stop_sequences was generated
+ * eos_token: The model generated an end-of-sentence (EOS) token
+ *
+ * stop_sequence: One of the sequence in stop_sequences was generated
  */
-type FinishReason = "length" | "eos_token" | "stop_sequence";
-interface PrefillToken {
+type TextGenerationFinishReason = "length" | "eos_token" | "stop_sequence";
+interface TextGenerationPrefillToken {
     id: number;
     logprob: number;
     /**
@@ -2756,9 +2961,12 @@ interface PrefillToken {
     text: string;
     [property: string]: unknown;
 }
-interface Token {
+/**
+ * Generated token.
+ */
+interface TextGenerationOutputToken {
     id: number;
-    logprob: number;
+    logprob?: number;
     /**
      * Whether or not that token is a special one
      */
@@ -3263,4 +3471,4 @@ declare namespace index {
   };
 }
 
-export { ALL_DISPLAY_MODEL_LIBRARY_KEYS, ALL_MODEL_LIBRARY_KEYS, AddedToken, AudioClassificationInput, AudioClassificationOutput, AudioClassificationOutputElement, AudioClassificationParameters, AutomaticSpeechRecognitionInput, AutomaticSpeechRecognitionOutput, AutomaticSpeechRecognitionOutputChunk, AutomaticSpeechRecognitionParameters, BoundingBox, ChatMessage, ClassificationOutputTransform$1 as ClassificationOutputTransform, DepthEstimationInput, DepthEstimationOutput, DocumentQuestionAnsweringInput, DocumentQuestionAnsweringInputData, DocumentQuestionAnsweringOutput, DocumentQuestionAnsweringOutputElement, DocumentQuestionAnsweringParameters, EarlyStoppingUnion$2 as EarlyStoppingUnion, ExampleRepo, FeatureExtractionInput, FeatureExtractionOutput, FillMaskInput, FillMaskOutput, FillMaskOutputElement, FillMaskParameters, FinishReason, GenerationParameters$2 as GenerationParameters, ImageClassificationInput, ImageClassificationOutput, ImageClassificationOutputElement, ImageClassificationParameters, ImageSegmentationInput, ImageSegmentationOutput, ImageSegmentationOutputElement, ImageSegmentationParameters, ImageSegmentationSubtask, ImageToImageInput, ImageToImageOutput, ImageToImageParameters, ImageToTextInput, ImageToTextOutput, ImageToTextParameters, InferenceDisplayability, LIBRARY_TASK_MAPPING_EXCLUDING_TRANSFORMERS, LibraryUiElement, MAPPING_DEFAULT_WIDGET, MODALITIES, MODALITY_LABELS, MODEL_LIBRARIES_UI_ELEMENTS, Modality, ModelData, ModelLibraryKey, ObjectDetectionInput, ObjectDetectionOutput, ObjectDetectionOutputElement, ObjectDetectionParameters, PIPELINE_DATA, PIPELINE_TYPES, PIPELINE_TYPES_SET, PipelineData, PipelineType, PrefillToken, QuestionAnsweringInput, QuestionAnsweringInputData, QuestionAnsweringOutput, QuestionAnsweringOutputElement, QuestionAnsweringParameters, SPECIAL_TOKENS_ATTRIBUTES, SUBTASK_TYPES, SentenceSimilarityInput, SentenceSimilarityInputData, SentenceSimilarityOutput, SpecialTokensMap, SummarizationInput, SummarizationOutput, TASKS_DATA, TASKS_MODEL_LIBRARIES, TableQuestionAnsweringInput, TableQuestionAnsweringInputData, TableQuestionAnsweringOutput, TableQuestionAnsweringOutputElement, TargetSize$1 as TargetSize, TaskData, TaskDataCustom, TaskDemo, TaskDemoEntry, Text2TextGenerationParameters, Text2TextGenerationTruncationStrategy, TextClassificationInput, TextClassificationOutput, TextClassificationOutputElement, TextClassificationParameters, TextGenerationInput, TextGenerationOutput, TextGenerationOutputDetails, TextGenerationParameters, TextGenerationSequenceDetails, TextToAudioParameters, TextToImageInput, TextToImageOutput, TextToImageParameters, TextToSpeechInput, TextToSpeechOutput, Token, TokenClassificationAggregationStrategy, TokenClassificationInput, TokenClassificationOutput, TokenClassificationOutputElement, TokenClassificationParameters, TokenizerConfig, TransformersInfo, TranslationInput, TranslationOutput, VideoClassificationInput, VideoClassificationOutput, VideoClassificationOutputElement, VideoClassificationParameters, VisualQuestionAnsweringInput, VisualQuestionAnsweringInputData, VisualQuestionAnsweringOutput, VisualQuestionAnsweringOutputElement, VisualQuestionAnsweringParameters, WidgetExample, WidgetExampleAssetAndPromptInput, WidgetExampleAssetAndTextInput, WidgetExampleAssetAndZeroShotInput, WidgetExampleAssetInput, WidgetExampleAttribute, WidgetExampleChatInput, WidgetExampleOutput, WidgetExampleOutputAnswerScore, WidgetExampleOutputLabels, WidgetExampleOutputText, WidgetExampleOutputUrl, WidgetExampleSentenceSimilarityInput, WidgetExampleStructuredDataInput, WidgetExampleTableDataInput, WidgetExampleTextAndContextInput, WidgetExampleTextAndTableInput, WidgetExampleTextInput, WidgetExampleZeroShotTextInput, WidgetType, WordBox, ZeroShotClassificationInput, ZeroShotClassificationInputData, ZeroShotClassificationOutput, ZeroShotClassificationOutputElement, ZeroShotClassificationParameters, ZeroShotImageClassificationInput, ZeroShotImageClassificationInputData, ZeroShotImageClassificationOutput, ZeroShotImageClassificationOutputElement, ZeroShotImageClassificationParameters, ZeroShotObjectDetectionInput, ZeroShotObjectDetectionInputData, ZeroShotObjectDetectionOutput, ZeroShotObjectDetectionOutputElement, index as snippets };
+export { ALL_DISPLAY_MODEL_LIBRARY_KEYS, ALL_MODEL_LIBRARY_KEYS, AddedToken, AudioClassificationInput, AudioClassificationOutput, AudioClassificationOutputElement, AudioClassificationParameters, AutomaticSpeechRecognitionInput, AutomaticSpeechRecognitionOutput, AutomaticSpeechRecognitionOutputChunk, AutomaticSpeechRecognitionParameters, BoundingBox, ChatCompletionFinishReason, ChatCompletionInput, ChatCompletionInputMessage, ChatCompletionOutput, ChatCompletionOutputChoice, ChatCompletionOutputChoiceMessage, ChatCompletionStreamOutput, ChatCompletionStreamOutputChoice, ChatCompletionStreamOutputDelta, ChatMessage, ClassificationOutputTransform$1 as ClassificationOutputTransform, DepthEstimationInput, DepthEstimationOutput, DocumentQuestionAnsweringInput, DocumentQuestionAnsweringInputData, DocumentQuestionAnsweringOutput, DocumentQuestionAnsweringOutputElement, DocumentQuestionAnsweringParameters, EarlyStoppingUnion$2 as EarlyStoppingUnion, ExampleRepo, FeatureExtractionInput, FeatureExtractionOutput, FillMaskInput, FillMaskOutput, FillMaskOutputElement, FillMaskParameters, GenerationParameters$2 as GenerationParameters, ImageClassificationInput, ImageClassificationOutput, ImageClassificationOutputElement, ImageClassificationParameters, ImageSegmentationInput, ImageSegmentationOutput, ImageSegmentationOutputElement, ImageSegmentationParameters, ImageSegmentationSubtask, ImageToImageInput, ImageToImageOutput, ImageToImageParameters, ImageToTextInput, ImageToTextOutput, ImageToTextParameters, InferenceDisplayability, LIBRARY_TASK_MAPPING_EXCLUDING_TRANSFORMERS, LibraryUiElement, MAPPING_DEFAULT_WIDGET, MODALITIES, MODALITY_LABELS, MODEL_LIBRARIES_UI_ELEMENTS, Modality, ModelData, ModelLibraryKey, ObjectDetectionInput, ObjectDetectionOutput, ObjectDetectionOutputElement, ObjectDetectionParameters, PIPELINE_DATA, PIPELINE_TYPES, PIPELINE_TYPES_SET, PipelineData, PipelineType, QuestionAnsweringInput, QuestionAnsweringInputData, QuestionAnsweringOutput, QuestionAnsweringOutputElement, QuestionAnsweringParameters, SPECIAL_TOKENS_ATTRIBUTES, SUBTASK_TYPES, SentenceSimilarityInput, SentenceSimilarityInputData, SentenceSimilarityOutput, SpecialTokensMap, SummarizationInput, SummarizationOutput, TASKS_DATA, TASKS_MODEL_LIBRARIES, TableQuestionAnsweringInput, TableQuestionAnsweringInputData, TableQuestionAnsweringOutput, TableQuestionAnsweringOutputElement, TargetSize$1 as TargetSize, TaskData, TaskDataCustom, TaskDemo, TaskDemoEntry, Text2TextGenerationParameters, Text2TextGenerationTruncationStrategy, TextClassificationInput, TextClassificationOutput, TextClassificationOutputElement, TextClassificationParameters, TextGenerationFinishReason, TextGenerationInput, TextGenerationOutput, TextGenerationOutputDetails, TextGenerationOutputSequenceDetails, TextGenerationOutputToken, TextGenerationParameters, TextGenerationPrefillToken, TextToAudioParameters, TextToImageInput, TextToImageOutput, TextToImageParameters, TextToSpeechInput, TextToSpeechOutput, TokenClassificationAggregationStrategy, TokenClassificationInput, TokenClassificationOutput, TokenClassificationOutputElement, TokenClassificationParameters, TokenizerConfig, TransformersInfo, TranslationInput, TranslationOutput, VideoClassificationInput, VideoClassificationOutput, VideoClassificationOutputElement, VideoClassificationParameters, VisualQuestionAnsweringInput, VisualQuestionAnsweringInputData, VisualQuestionAnsweringOutput, VisualQuestionAnsweringOutputElement, VisualQuestionAnsweringParameters, WidgetExample, WidgetExampleAssetAndPromptInput, WidgetExampleAssetAndTextInput, WidgetExampleAssetAndZeroShotInput, WidgetExampleAssetInput, WidgetExampleAttribute, WidgetExampleChatInput, WidgetExampleOutput, WidgetExampleOutputAnswerScore, WidgetExampleOutputLabels, WidgetExampleOutputText, WidgetExampleOutputUrl, WidgetExampleSentenceSimilarityInput, WidgetExampleStructuredDataInput, WidgetExampleTableDataInput, WidgetExampleTextAndContextInput, WidgetExampleTextAndTableInput, WidgetExampleTextInput, WidgetExampleZeroShotTextInput, WidgetType, WordBox, ZeroShotClassificationInput, ZeroShotClassificationInputData, ZeroShotClassificationOutput, ZeroShotClassificationOutputElement, ZeroShotClassificationParameters, ZeroShotImageClassificationInput, ZeroShotImageClassificationInputData, ZeroShotImageClassificationOutput, ZeroShotImageClassificationOutputElement, ZeroShotImageClassificationParameters, ZeroShotObjectDetectionInput, ZeroShotObjectDetectionInputData, ZeroShotObjectDetectionOutput, ZeroShotObjectDetectionOutputElement, index as snippets };

package/dist/index.js

@@ -4048,6 +4048,11 @@ var flair = (model) => [
 
 tagger = SequenceTagger.load("${model.id}")`
 ];
+var gliner = (model) => [
+  `from gliner import GLiNER
+
+model = GLiNER.from_pretrained("${model.id}")`
+];
 var keras = (model) => [
   `from huggingface_hub import from_pretrained_keras
 
@@ -4394,6 +4399,11 @@ var mlx = (model) => [
 export HF_HUB_ENABLE_HF_TRANS: string[]FER=1
 huggingface-cli download --local-dir ${nameWithoutNamespace(model.id)} ${model.id}`
 ];
+var mlxim = (model) => [
+  `from mlxim.model import create_model
+
+model = create_model(${model.id})`
+];
 var nemo = (model) => {
   let command = void 0;
   if (model.tags?.includes("automatic-speech-recognition")) {
@@ -4543,6 +4553,25 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
       term: { path: "pytorch_model.bin" }
     }
   },
+  gliner: {
+    prettyLabel: "GLiNER",
+    repoName: "GLiNER",
+    repoUrl: "https://github.com/urchade/GLiNER",
+    snippets: gliner,
+    filter: false,
+    countDownloads: {
+      term: { 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"] }
+    }
+  },
   keras: {
     prettyLabel: "Keras",
     repoName: "Keras",
@@ -4578,6 +4607,15 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     snippets: mlx,
     filter: true
   },
+  "mlx-image": {
+    prettyLabel: "mlx-image",
+    repoName: "mlx-image",
+    repoUrl: "https://github.com/riccardomusmeci/mlx-image",
+    docsUrl: "https://huggingface.co/docs/hub/mlx-image",
+    snippets: mlxim,
+    filter: false,
+    countDownloads: { term: { path: "model.safetensors" } }
+  },
   nemo: {
     prettyLabel: "NeMo",
     repoName: "NeMo",
@@ -4758,6 +4796,14 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     countDownloads: {
       wildcard: { path: "*.sentis" }
     }
+  },
+  whisperkit: {
+    prettyLabel: "WhisperKit",
+    repoName: "WhisperKit",
+    repoUrl: "https://github.com/argmaxinc/WhisperKit",
+    countDownloads: {
+      wildcard: { path: "*/model.mil" }
+    }
   }
 };
 var ALL_MODEL_LIBRARY_KEYS = Object.keys(MODEL_LIBRARIES_UI_ELEMENTS);

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@huggingface/tasks",
   "packageManager": "pnpm@8.10.5",
-  "version": "0.5.2",
+  "version": "0.6.1",
   "description": "List of ML tasks for huggingface.co/tasks",
   "repository": "https://github.com/huggingface/huggingface.js.git",
   "publishConfig": {
@@ -33,7 +33,7 @@
   "license": "MIT",
   "devDependencies": {
     "@types/node": "^20.11.5",
-    "quicktype-core": "https://github.com/huggingface/quicktype/raw/pack-18.0.15/packages/quicktype-core/quicktype-core-18.0.15.tgz"
+    "quicktype-core": "https://github.com/huggingface/quicktype/raw/pack-18.0.17/packages/quicktype-core/quicktype-core-18.0.17.tgz"
   },
   "scripts": {
     "lint": "eslint --quiet --fix --ext .cjs,.ts .",

package/src/model-libraries-snippets.ts

@@ -140,6 +140,12 @@ export const flair = (model: ModelData): string[] => [
 tagger = SequenceTagger.load("${model.id}")`,
 ];
 
+export const gliner = (model: ModelData): string[] => [
+	`from gliner import GLiNER
+
+model = GLiNER.from_pretrained("${model.id}")`,
+];
+
 export const keras = (model: ModelData): string[] => [
 	`from huggingface_hub import from_pretrained_keras
 
@@ -532,6 +538,12 @@ export HF_HUB_ENABLE_HF_TRANS: string[]FER=1
 huggingface-cli download --local-dir ${nameWithoutNamespace(model.id)} ${model.id}`,
 ];
 
+export const mlxim = (model: ModelData): string[] => [
+	`from mlxim.model import create_model
+
+model = create_model(${model.id})`,
+];
+
 export const nemo = (model: ModelData): string[] => {
 	let command: string[] | undefined = undefined;
 	// Resolve the tag to a nemo domain/sub-domain

package/src/model-libraries.ts

@@ -157,6 +157,25 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 			term: { path: "pytorch_model.bin" },
 		},
 	},
+	gliner: {
+		prettyLabel: "GLiNER",
+		repoName: "GLiNER",
+		repoUrl: "https://github.com/urchade/GLiNER",
+		snippets: snippets.gliner,
+		filter: false,
+		countDownloads: {
+			term: { 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"] },
+		},
+	},
 	keras: {
 		prettyLabel: "Keras",
 		repoName: "Keras",
@@ -192,6 +211,15 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 		snippets: snippets.mlx,
 		filter: true,
 	},
+	"mlx-image": {
+		prettyLabel: "mlx-image",
+		repoName: "mlx-image",
+		repoUrl: "https://github.com/riccardomusmeci/mlx-image",
+		docsUrl: "https://huggingface.co/docs/hub/mlx-image",
+		snippets: snippets.mlxim,
+		filter: false,
+		countDownloads: { term: { path: "model.safetensors" } },
+	},
 	nemo: {
 		prettyLabel: "NeMo",
 		repoName: "NeMo",
@@ -373,6 +401,14 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 			wildcard: { path: "*.sentis" },
 		},
 	},
+	whisperkit: {
+		prettyLabel: "WhisperKit",
+		repoName: "WhisperKit",
+		repoUrl: "https://github.com/argmaxinc/WhisperKit",
+		countDownloads: {
+			wildcard: { path: "*/model.mil" },
+		},
+	},
 } satisfies Record<string, LibraryUiElement>;
 
 export type ModelLibraryKey = keyof typeof MODEL_LIBRARIES_UI_ELEMENTS;