@huggingface/tasks 0.13.1 → 0.13.2

package/dist/commonjs/model-libraries-snippets.js

@@ -0,0 +1,1093 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.transformersJS = exports.transformers = exports.speechbrain = exports.stanza = exports.span_marker = exports.spacy = exports.setfit = exports.sentenceTransformers = exports.sampleFactory = exports.sam2 = exports.fastai = exports.stable_audio_tools = exports.sklearn = exports.seed_story = exports.saelens = exports.timm = exports.tensorflowtts = exports.relik = exports.pyannote_audio = exports.pyannote_audio_pipeline = exports.paddlenlp = exports.open_clip = exports.mesh_anything = exports.mars5_tts = exports.mamba_ssm = exports.tf_keras = exports.llama_cpp_python = exports.keras_hub = exports.keras_nlp = exports.keras = exports.htrflow = exports.gliner = exports.flair = exports.fairseq = exports.espnet = exports.espnetASR = exports.espnetTTS = exports.edsnlp = exports.cartesia_mlx = exports.cartesia_pytorch = exports.diffusionkit = exports.diffusers = exports.depth_pro = exports.depth_anything_v2 = exports.bm25s = exports.bertopic = exports.audioseal = exports.asteroid = exports.allennlp = exports.adapters = void 0;
+exports.hezar = exports.threedtopia_xl = exports.whisperkit = exports.audiocraft = exports.pythae = exports.pxia = exports.nemo = exports.model2vec = exports.mlxim = exports.mlx = exports.birefnet = exports.yolov10 = exports.chattts = exports.voicecraft = exports.vfimamba = exports.sentis = exports.mlAgents = exports.stableBaselines3 = exports.fasttext = exports.peft = void 0;
+const library_to_tasks_js_1 = require("./library-to-tasks.js");
+const TAG_CUSTOM_CODE = "custom_code";
+function nameWithoutNamespace(modelId) {
+    const splitted = modelId.split("/");
+    return splitted.length === 1 ? splitted[0] : splitted[1];
+}
+const escapeStringForJson = (str) => JSON.stringify(str).slice(1, -1); // slice is needed to remove surrounding quotes added by JSON.stringify
+//#region snippets
+const adapters = (model) => [
+    `from adapters import AutoAdapterModel
+
+model = AutoAdapterModel.from_pretrained("${model.config?.adapter_transformers?.model_name}")
+model.load_adapter("${model.id}", set_active=True)`,
+];
+exports.adapters = adapters;
+const allennlpUnknown = (model) => [
+    `import allennlp_models
+from allennlp.predictors.predictor import Predictor
+
+predictor = Predictor.from_path("hf://${model.id}")`,
+];
+const allennlpQuestionAnswering = (model) => [
+    `import allennlp_models
+from allennlp.predictors.predictor import Predictor
+
+predictor = Predictor.from_path("hf://${model.id}")
+predictor_input = {"passage": "My name is Wolfgang and I live in Berlin", "question": "Where do I live?"}
+predictions = predictor.predict_json(predictor_input)`,
+];
+const allennlp = (model) => {
+    if (model.tags.includes("question-answering")) {
+        return allennlpQuestionAnswering(model);
+    }
+    return allennlpUnknown(model);
+};
+exports.allennlp = allennlp;
+const asteroid = (model) => [
+    `from asteroid.models import BaseModel
+
+model = BaseModel.from_pretrained("${model.id}")`,
+];
+exports.asteroid = asteroid;
+const 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];
+};
+exports.audioseal = audioseal;
+function get_base_diffusers_model(model) {
+    return model.cardData?.base_model?.toString() ?? "fill-in-base-model";
+}
+function get_prompt_from_diffusers_model(model) {
+    const prompt = model.widgetData?.[0]?.text ?? model.cardData?.instance_prompt;
+    if (prompt) {
+        return escapeStringForJson(prompt);
+    }
+}
+const bertopic = (model) => [
+    `from bertopic import BERTopic
+
+model = BERTopic.load("${model.id}")`,
+];
+exports.bertopic = bertopic;
+const bm25s = (model) => [
+    `from bm25s.hf import BM25HF
+
+retriever = BM25HF.load_from_hub("${model.id}")`,
+];
+exports.bm25s = bm25s;
+const depth_anything_v2 = (model) => {
+    let encoder;
+    let features;
+    let out_channels;
+    encoder = "<ENCODER>";
+    features = "<NUMBER_OF_FEATURES>";
+    out_channels = "<OUT_CHANNELS>";
+    if (model.id === "depth-anything/Depth-Anything-V2-Small") {
+        encoder = "vits";
+        features = "64";
+        out_channels = "[48, 96, 192, 384]";
+    }
+    else if (model.id === "depth-anything/Depth-Anything-V2-Base") {
+        encoder = "vitb";
+        features = "128";
+        out_channels = "[96, 192, 384, 768]";
+    }
+    else if (model.id === "depth-anything/Depth-Anything-V2-Large") {
+        encoder = "vitl";
+        features = "256";
+        out_channels = "[256, 512, 1024, 1024";
+    }
+    return [
+        `
+# Install from https://github.com/DepthAnything/Depth-Anything-V2
+
+# Load the model and infer depth from an image
+import cv2
+import torch
+
+from depth_anything_v2.dpt import DepthAnythingV2
+
+# instantiate the model
+model = DepthAnythingV2(encoder="${encoder}", features=${features}, out_channels=${out_channels})
+
+# load the weights
+filepath = hf_hub_download(repo_id="${model.id}", filename="depth_anything_v2_${encoder}.pth", repo_type="model")
+state_dict = torch.load(filepath, map_location="cpu")
+model.load_state_dict(state_dict).eval()
+
+raw_img = cv2.imread("your/image/path")
+depth = model.infer_image(raw_img) # HxW raw depth map in numpy
+    `,
+    ];
+};
+exports.depth_anything_v2 = depth_anything_v2;
+const depth_pro = (model) => {
+    const installSnippet = `# Download checkpoint
+pip install huggingface-hub
+huggingface-cli download --local-dir checkpoints ${model.id}`;
+    const inferenceSnippet = `import depth_pro
+
+# Load model and preprocessing transform
+model, transform = depth_pro.create_model_and_transforms()
+model.eval()
+
+# Load and preprocess an image.
+image, _, f_px = depth_pro.load_rgb("example.png")
+image = transform(image)
+
+# Run inference.
+prediction = model.infer(image, f_px=f_px)
+
+# Results: 1. Depth in meters
+depth = prediction["depth"]
+# Results: 2. Focal length in pixels
+focallength_px = prediction["focallength_px"]`;
+    return [installSnippet, inferenceSnippet];
+};
+exports.depth_pro = depth_pro;
+const diffusersDefaultPrompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k";
+const diffusers_default = (model) => [
+    `from diffusers import DiffusionPipeline
+
+pipe = DiffusionPipeline.from_pretrained("${model.id}")
+
+prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersDefaultPrompt}"
+image = pipe(prompt).images[0]`,
+];
+const diffusers_controlnet = (model) => [
+    `from diffusers import ControlNetModel, StableDiffusionControlNetPipeline
+
+controlnet = ControlNetModel.from_pretrained("${model.id}")
+pipe = StableDiffusionControlNetPipeline.from_pretrained(
+	"${get_base_diffusers_model(model)}", controlnet=controlnet
+)`,
+];
+const diffusers_lora = (model) => [
+    `from diffusers import DiffusionPipeline
+
+pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}")
+pipe.load_lora_weights("${model.id}")
+
+prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersDefaultPrompt}"
+image = pipe(prompt).images[0]`,
+];
+const diffusers_textual_inversion = (model) => [
+    `from diffusers import DiffusionPipeline
+
+pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}")
+pipe.load_textual_inversion("${model.id}")`,
+];
+const diffusers = (model) => {
+    if (model.tags.includes("controlnet")) {
+        return diffusers_controlnet(model);
+    }
+    else if (model.tags.includes("lora")) {
+        return diffusers_lora(model);
+    }
+    else if (model.tags.includes("textual_inversion")) {
+        return diffusers_textual_inversion(model);
+    }
+    else {
+        return diffusers_default(model);
+    }
+};
+exports.diffusers = diffusers;
+const diffusionkit = (model) => {
+    const sd3Snippet = `# Pipeline for Stable Diffusion 3
+from diffusionkit.mlx import DiffusionPipeline
+
+pipeline = DiffusionPipeline(
+	shift=3.0,
+	use_t5=False,
+	model_version=${model.id},
+	low_memory_mode=True,
+	a16=True,
+	w16=True,
+)`;
+    const fluxSnippet = `# Pipeline for Flux
+from diffusionkit.mlx import FluxPipeline
+
+pipeline = FluxPipeline(
+  shift=1.0,
+  model_version=${model.id},
+  low_memory_mode=True,
+  a16=True,
+  w16=True,
+)`;
+    const generateSnippet = `# Image Generation
+HEIGHT = 512
+WIDTH = 512
+NUM_STEPS = ${model.tags.includes("flux") ? 4 : 50}
+CFG_WEIGHT = ${model.tags.includes("flux") ? 0 : 5}
+
+image, _ = pipeline.generate_image(
+  "a photo of a cat",
+  cfg_weight=CFG_WEIGHT,
+  num_steps=NUM_STEPS,
+  latent_size=(HEIGHT // 8, WIDTH // 8),
+)`;
+    const pipelineSnippet = model.tags.includes("flux") ? fluxSnippet : sd3Snippet;
+    return [pipelineSnippet, generateSnippet];
+};
+exports.diffusionkit = diffusionkit;
+const cartesia_pytorch = (model) => [
+    `# pip install --no-binary :all: cartesia-pytorch
+from cartesia_pytorch import ReneLMHeadModel
+from transformers import AutoTokenizer
+
+model = ReneLMHeadModel.from_pretrained("${model.id}")
+tokenizer = AutoTokenizer.from_pretrained("allenai/OLMo-1B-hf")
+
+in_message = ["Rene Descartes was"]
+inputs = tokenizer(in_message, return_tensors="pt")
+
+outputs = model.generate(inputs.input_ids, max_length=50, top_k=100, top_p=0.99)
+out_message = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
+
+print(out_message)
+)`,
+];
+exports.cartesia_pytorch = cartesia_pytorch;
+const cartesia_mlx = (model) => [
+    `import mlx.core as mx
+import cartesia_mlx as cmx
+
+model = cmx.from_pretrained("${model.id}")
+model.set_dtype(mx.float32)   
+
+prompt = "Rene Descartes was"
+
+for text in model.generate(
+    prompt,
+    max_tokens=500,
+    eval_every_n=5,
+    verbose=True,
+    top_p=0.99,
+    temperature=0.85,
+):
+    print(text, end="", flush=True)
+`,
+];
+exports.cartesia_mlx = cartesia_mlx;
+const edsnlp = (model) => {
+    const packageName = nameWithoutNamespace(model.id).replaceAll("-", "_");
+    return [
+        `# Load it from the Hub directly
+import edsnlp
+nlp = edsnlp.load("${model.id}")
+`,
+        `# Or install it as a package
+!pip install git+https://huggingface.co/${model.id}
+
+# and import it as a module
+import ${packageName}
+
+nlp = ${packageName}.load()  # or edsnlp.load("${packageName}")
+`,
+    ];
+};
+exports.edsnlp = edsnlp;
+const espnetTTS = (model) => [
+    `from espnet2.bin.tts_inference import Text2Speech
+
+model = Text2Speech.from_pretrained("${model.id}")
+
+speech, *_ = model("text to generate speech from")`,
+];
+exports.espnetTTS = espnetTTS;
+const espnetASR = (model) => [
+    `from espnet2.bin.asr_inference import Speech2Text
+
+model = Speech2Text.from_pretrained(
+  "${model.id}"
+)
+
+speech, rate = soundfile.read("speech.wav")
+text, *_ = model(speech)[0]`,
+];
+exports.espnetASR = espnetASR;
+const espnetUnknown = () => [`unknown model type (must be text-to-speech or automatic-speech-recognition)`];
+const espnet = (model) => {
+    if (model.tags.includes("text-to-speech")) {
+        return (0, exports.espnetTTS)(model);
+    }
+    else if (model.tags.includes("automatic-speech-recognition")) {
+        return (0, exports.espnetASR)(model);
+    }
+    return espnetUnknown();
+};
+exports.espnet = espnet;
+const fairseq = (model) => [
+    `from fairseq.checkpoint_utils import load_model_ensemble_and_task_from_hf_hub
+
+models, cfg, task = load_model_ensemble_and_task_from_hf_hub(
+    "${model.id}"
+)`,
+];
+exports.fairseq = fairseq;
+const flair = (model) => [
+    `from flair.models import SequenceTagger
+
+tagger = SequenceTagger.load("${model.id}")`,
+];
+exports.flair = flair;
+const gliner = (model) => [
+    `from gliner import GLiNER
+
+model = GLiNER.from_pretrained("${model.id}")`,
+];
+exports.gliner = gliner;
+const htrflow = (model) => [
+    `# CLI usage
+# see docs: https://ai-riksarkivet.github.io/htrflow/latest/getting_started/quick_start.html
+htrflow pipeline <path/to/pipeline.yaml> <path/to/image>`,
+    `# Python usage
+from htrflow.pipeline.pipeline import Pipeline
+from htrflow.pipeline.steps import Task
+from htrflow.models.framework.model import ModelClass
+
+pipeline = Pipeline(
+    [
+        Task(
+            ModelClass, {"model": "${model.id}"}, {}
+        ),
+    ])`,
+];
+exports.htrflow = htrflow;
+const keras = (model) => [
+    `# Available backend options are: "jax", "torch", "tensorflow".
+import os
+os.environ["KERAS_BACKEND"] = "jax"
+	
+import keras
+
+model = keras.saving.load_model("hf://${model.id}")
+`,
+];
+exports.keras = keras;
+const keras_nlp = (model) => [
+    `# Available backend options are: "jax", "torch", "tensorflow".
+import os
+os.environ["KERAS_BACKEND"] = "jax"
+
+import keras_nlp
+
+tokenizer = keras_nlp.models.Tokenizer.from_preset("hf://${model.id}")
+backbone = keras_nlp.models.Backbone.from_preset("hf://${model.id}")
+`,
+];
+exports.keras_nlp = keras_nlp;
+const keras_hub = (model) => [
+    `# Available backend options are: "jax", "torch", "tensorflow".
+import os
+os.environ["KERAS_BACKEND"] = "jax"
+
+import keras_hub
+
+# Load a task-specific model (*replace CausalLM with your task*)
+model = keras_hub.models.CausalLM.from_preset("hf://${model.id}", dtype="bfloat16")
+
+# Possible tasks are CausalLM, TextToImage, ImageClassifier, ...
+# full list here: https://keras.io/api/keras_hub/models/#api-documentation
+`,
+];
+exports.keras_hub = keras_hub;
+const llama_cpp_python = (model) => [
+    `from llama_cpp import Llama
+
+llm = Llama.from_pretrained(
+	repo_id="${model.id}",
+	filename="{{GGUF_FILE}}",
+)
+
+llm.create_chat_completion(
+	messages = [
+		{
+			"role": "user",
+			"content": "What is the capital of France?"
+		}
+	]
+)`,
+];
+exports.llama_cpp_python = llama_cpp_python;
+const tf_keras = (model) => [
+    `# Note: 'keras<3.x' or 'tf_keras' must be installed (legacy)
+# See https://github.com/keras-team/tf-keras for more details.
+from huggingface_hub import from_pretrained_keras
+
+model = from_pretrained_keras("${model.id}")
+`,
+];
+exports.tf_keras = tf_keras;
+const mamba_ssm = (model) => [
+    `from mamba_ssm import MambaLMHeadModel
+
+model = MambaLMHeadModel.from_pretrained("${model.id}")`,
+];
+exports.mamba_ssm = mamba_ssm;
+const mars5_tts = (model) => [
+    `# Install from https://github.com/Camb-ai/MARS5-TTS
+
+from inference import Mars5TTS
+mars5 = Mars5TTS.from_pretrained("${model.id}")`,
+];
+exports.mars5_tts = mars5_tts;
+const mesh_anything = () => [
+    `# Install from https://github.com/buaacyw/MeshAnything.git
+
+from MeshAnything.models.meshanything import MeshAnything
+
+# refer to https://github.com/buaacyw/MeshAnything/blob/main/main.py#L91 on how to define args
+# and https://github.com/buaacyw/MeshAnything/blob/main/app.py regarding usage
+model = MeshAnything(args)`,
+];
+exports.mesh_anything = mesh_anything;
+const open_clip = (model) => [
+    `import open_clip
+
+model, preprocess_train, preprocess_val = open_clip.create_model_and_transforms('hf-hub:${model.id}')
+tokenizer = open_clip.get_tokenizer('hf-hub:${model.id}')`,
+];
+exports.open_clip = open_clip;
+const paddlenlp = (model) => {
+    if (model.config?.architectures?.[0]) {
+        const architecture = model.config.architectures[0];
+        return [
+            [
+                `from paddlenlp.transformers import AutoTokenizer, ${architecture}`,
+                "",
+                `tokenizer = AutoTokenizer.from_pretrained("${model.id}", from_hf_hub=True)`,
+                `model = ${architecture}.from_pretrained("${model.id}", from_hf_hub=True)`,
+            ].join("\n"),
+        ];
+    }
+    else {
+        return [
+            [
+                `# ⚠️ Type of model unknown`,
+                `from paddlenlp.transformers import AutoTokenizer, AutoModel`,
+                "",
+                `tokenizer = AutoTokenizer.from_pretrained("${model.id}", from_hf_hub=True)`,
+                `model = AutoModel.from_pretrained("${model.id}", from_hf_hub=True)`,
+            ].join("\n"),
+        ];
+    }
+};
+exports.paddlenlp = paddlenlp;
+const pyannote_audio_pipeline = (model) => [
+    `from pyannote.audio import Pipeline
+  
+pipeline = Pipeline.from_pretrained("${model.id}")
+
+# inference on the whole file
+pipeline("file.wav")
+
+# inference on an excerpt
+from pyannote.core import Segment
+excerpt = Segment(start=2.0, end=5.0)
+
+from pyannote.audio import Audio
+waveform, sample_rate = Audio().crop("file.wav", excerpt)
+pipeline({"waveform": waveform, "sample_rate": sample_rate})`,
+];
+exports.pyannote_audio_pipeline = pyannote_audio_pipeline;
+const pyannote_audio_model = (model) => [
+    `from pyannote.audio import Model, Inference
+
+model = Model.from_pretrained("${model.id}")
+inference = Inference(model)
+
+# inference on the whole file
+inference("file.wav")
+
+# inference on an excerpt
+from pyannote.core import Segment
+excerpt = Segment(start=2.0, end=5.0)
+inference.crop("file.wav", excerpt)`,
+];
+const pyannote_audio = (model) => {
+    if (model.tags.includes("pyannote-audio-pipeline")) {
+        return (0, exports.pyannote_audio_pipeline)(model);
+    }
+    return pyannote_audio_model(model);
+};
+exports.pyannote_audio = pyannote_audio;
+const relik = (model) => [
+    `from relik import Relik
+ 
+relik = Relik.from_pretrained("${model.id}")`,
+];
+exports.relik = relik;
+const tensorflowttsTextToMel = (model) => [
+    `from tensorflow_tts.inference import AutoProcessor, TFAutoModel
+
+processor = AutoProcessor.from_pretrained("${model.id}")
+model = TFAutoModel.from_pretrained("${model.id}")
+`,
+];
+const tensorflowttsMelToWav = (model) => [
+    `from tensorflow_tts.inference import TFAutoModel
+
+model = TFAutoModel.from_pretrained("${model.id}")
+audios = model.inference(mels)
+`,
+];
+const tensorflowttsUnknown = (model) => [
+    `from tensorflow_tts.inference import TFAutoModel
+
+model = TFAutoModel.from_pretrained("${model.id}")
+`,
+];
+const tensorflowtts = (model) => {
+    if (model.tags.includes("text-to-mel")) {
+        return tensorflowttsTextToMel(model);
+    }
+    else if (model.tags.includes("mel-to-wav")) {
+        return tensorflowttsMelToWav(model);
+    }
+    return tensorflowttsUnknown(model);
+};
+exports.tensorflowtts = tensorflowtts;
+const timm = (model) => [
+    `import timm
+
+model = timm.create_model("hf_hub:${model.id}", pretrained=True)`,
+];
+exports.timm = timm;
+const saelens = ( /* model: ModelData */) => [
+    `# pip install sae-lens
+from sae_lens import SAE
+
+sae, cfg_dict, sparsity = SAE.from_pretrained(
+    release = "RELEASE_ID", # e.g., "gpt2-small-res-jb". See other options in https://github.com/jbloomAus/SAELens/blob/main/sae_lens/pretrained_saes.yaml
+    sae_id = "SAE_ID", # e.g., "blocks.8.hook_resid_pre". Won't always be a hook point
+)`,
+];
+exports.saelens = saelens;
+const seed_story = () => [
+    `# seed_story_cfg_path refers to 'https://github.com/TencentARC/SEED-Story/blob/master/configs/clm_models/agent_7b_sft.yaml'
+# llm_cfg_path refers to 'https://github.com/TencentARC/SEED-Story/blob/master/configs/clm_models/llama2chat7b_lora.yaml'
+from omegaconf import OmegaConf
+import hydra
+
+# load Llama2
+llm_cfg = OmegaConf.load(llm_cfg_path)
+llm = hydra.utils.instantiate(llm_cfg, torch_dtype="fp16")
+
+# initialize seed_story
+seed_story_cfg = OmegaConf.load(seed_story_cfg_path)
+seed_story = hydra.utils.instantiate(seed_story_cfg, llm=llm) `,
+];
+exports.seed_story = seed_story;
+const skopsPickle = (model, modelFile) => {
+    return [
+        `import joblib
+from skops.hub_utils import download
+download("${model.id}", "path_to_folder")
+model = joblib.load(
+	"${modelFile}"
+)
+# only load pickle files from sources you trust
+# read more about it here https://skops.readthedocs.io/en/stable/persistence.html`,
+    ];
+};
+const skopsFormat = (model, modelFile) => {
+    return [
+        `from skops.hub_utils import download
+from skops.io import load
+download("${model.id}", "path_to_folder")
+# make sure model file is in skops format
+# if model is a pickle file, make sure it's from a source you trust
+model = load("path_to_folder/${modelFile}")`,
+    ];
+};
+const skopsJobLib = (model) => {
+    return [
+        `from huggingface_hub import hf_hub_download
+import joblib
+model = joblib.load(
+	hf_hub_download("${model.id}", "sklearn_model.joblib")
+)
+# only load pickle files from sources you trust
+# read more about it here https://skops.readthedocs.io/en/stable/persistence.html`,
+    ];
+};
+const sklearn = (model) => {
+    if (model.tags.includes("skops")) {
+        const skopsmodelFile = model.config?.sklearn?.model?.file;
+        const skopssaveFormat = model.config?.sklearn?.model_format;
+        if (!skopsmodelFile) {
+            return [`# ⚠️ Model filename not specified in config.json`];
+        }
+        if (skopssaveFormat === "pickle") {
+            return skopsPickle(model, skopsmodelFile);
+        }
+        else {
+            return skopsFormat(model, skopsmodelFile);
+        }
+    }
+    else {
+        return skopsJobLib(model);
+    }
+};
+exports.sklearn = sklearn;
+const stable_audio_tools = (model) => [
+    `import torch
+import torchaudio
+from einops import rearrange
+from stable_audio_tools import get_pretrained_model
+from stable_audio_tools.inference.generation import generate_diffusion_cond
+
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+# Download model
+model, model_config = get_pretrained_model("${model.id}")
+sample_rate = model_config["sample_rate"]
+sample_size = model_config["sample_size"]
+
+model = model.to(device)
+
+# Set up text and timing conditioning
+conditioning = [{
+	"prompt": "128 BPM tech house drum loop",
+}]
+
+# Generate stereo audio
+output = generate_diffusion_cond(
+	model,
+	conditioning=conditioning,
+	sample_size=sample_size,
+	device=device
+)
+
+# Rearrange audio batch to a single sequence
+output = rearrange(output, "b d n -> d (b n)")
+
+# Peak normalize, clip, convert to int16, and save to file
+output = output.to(torch.float32).div(torch.max(torch.abs(output))).clamp(-1, 1).mul(32767).to(torch.int16).cpu()
+torchaudio.save("output.wav", output, sample_rate)`,
+];
+exports.stable_audio_tools = stable_audio_tools;
+const fastai = (model) => [
+    `from huggingface_hub import from_pretrained_fastai
+
+learn = from_pretrained_fastai("${model.id}")`,
+];
+exports.fastai = fastai;
+const sam2 = (model) => {
+    const image_predictor = `# Use SAM2 with images
+import torch
+from sam2.sam2_image_predictor import SAM2ImagePredictor
+
+predictor = SAM2ImagePredictor.from_pretrained(${model.id})
+
+with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16):
+    predictor.set_image(<your_image>)
+    masks, _, _ = predictor.predict(<input_prompts>)`;
+    const video_predictor = `# Use SAM2 with videos
+import torch
+from sam2.sam2_video_predictor import SAM2VideoPredictor
+	
+predictor = SAM2VideoPredictor.from_pretrained(${model.id})
+
+with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16):
+    state = predictor.init_state(<your_video>)
+
+    # add new prompts and instantly get the output on the same frame
+    frame_idx, object_ids, masks = predictor.add_new_points(state, <your_prompts>):
+
+    # propagate the prompts to get masklets throughout the video
+    for frame_idx, object_ids, masks in predictor.propagate_in_video(state):
+        ...`;
+    return [image_predictor, video_predictor];
+};
+exports.sam2 = sam2;
+const sampleFactory = (model) => [
+    `python -m sample_factory.huggingface.load_from_hub -r ${model.id} -d ./train_dir`,
+];
+exports.sampleFactory = sampleFactory;
+function get_widget_examples_from_st_model(model) {
+    const widgetExample = model.widgetData?.[0];
+    if (widgetExample) {
+        return [widgetExample.source_sentence, ...widgetExample.sentences];
+    }
+}
+const sentenceTransformers = (model) => {
+    const remote_code_snippet = model.tags.includes(TAG_CUSTOM_CODE) ? ", trust_remote_code=True" : "";
+    const exampleSentences = get_widget_examples_from_st_model(model) ?? [
+        "The weather is lovely today.",
+        "It's so sunny outside!",
+        "He drove to the stadium.",
+    ];
+    return [
+        `from sentence_transformers import SentenceTransformer
+
+model = SentenceTransformer("${model.id}"${remote_code_snippet})
+
+sentences = ${JSON.stringify(exampleSentences, null, 4)}
+embeddings = model.encode(sentences)
+
+similarities = model.similarity(embeddings, embeddings)
+print(similarities.shape)
+# [${exampleSentences.length}, ${exampleSentences.length}]`,
+    ];
+};
+exports.sentenceTransformers = sentenceTransformers;
+const setfit = (model) => [
+    `from setfit import SetFitModel
+
+model = SetFitModel.from_pretrained("${model.id}")`,
+];
+exports.setfit = setfit;
+const spacy = (model) => [
+    `!pip install https://huggingface.co/${model.id}/resolve/main/${nameWithoutNamespace(model.id)}-any-py3-none-any.whl
+
+# Using spacy.load().
+import spacy
+nlp = spacy.load("${nameWithoutNamespace(model.id)}")
+
+# Importing as module.
+import ${nameWithoutNamespace(model.id)}
+nlp = ${nameWithoutNamespace(model.id)}.load()`,
+];
+exports.spacy = spacy;
+const span_marker = (model) => [
+    `from span_marker import SpanMarkerModel
+
+model = SpanMarkerModel.from_pretrained("${model.id}")`,
+];
+exports.span_marker = span_marker;
+const stanza = (model) => [
+    `import stanza
+
+stanza.download("${nameWithoutNamespace(model.id).replace("stanza-", "")}")
+nlp = stanza.Pipeline("${nameWithoutNamespace(model.id).replace("stanza-", "")}")`,
+];
+exports.stanza = stanza;
+const speechBrainMethod = (speechbrainInterface) => {
+    switch (speechbrainInterface) {
+        case "EncoderClassifier":
+            return "classify_file";
+        case "EncoderDecoderASR":
+        case "EncoderASR":
+            return "transcribe_file";
+        case "SpectralMaskEnhancement":
+            return "enhance_file";
+        case "SepformerSeparation":
+            return "separate_file";
+        default:
+            return undefined;
+    }
+};
+const speechbrain = (model) => {
+    const speechbrainInterface = model.config?.speechbrain?.speechbrain_interface;
+    if (speechbrainInterface === undefined) {
+        return [`# interface not specified in config.json`];
+    }
+    const speechbrainMethod = speechBrainMethod(speechbrainInterface);
+    if (speechbrainMethod === undefined) {
+        return [`# interface in config.json invalid`];
+    }
+    return [
+        `from speechbrain.pretrained import ${speechbrainInterface}
+model = ${speechbrainInterface}.from_hparams(
+  "${model.id}"
+)
+model.${speechbrainMethod}("file.wav")`,
+    ];
+};
+exports.speechbrain = speechbrain;
+const transformers = (model) => {
+    const info = model.transformersInfo;
+    if (!info) {
+        return [`# ⚠️ Type of model unknown`];
+    }
+    const remote_code_snippet = model.tags.includes(TAG_CUSTOM_CODE) ? ", trust_remote_code=True" : "";
+    let autoSnippet;
+    if (info.processor) {
+        const varName = info.processor === "AutoTokenizer"
+            ? "tokenizer"
+            : info.processor === "AutoFeatureExtractor"
+                ? "extractor"
+                : "processor";
+        autoSnippet = [
+            "# Load model directly",
+            `from transformers import ${info.processor}, ${info.auto_model}`,
+            "",
+            `${varName} = ${info.processor}.from_pretrained("${model.id}"` + remote_code_snippet + ")",
+            `model = ${info.auto_model}.from_pretrained("${model.id}"` + remote_code_snippet + ")",
+        ].join("\n");
+    }
+    else {
+        autoSnippet = [
+            "# Load model directly",
+            `from transformers import ${info.auto_model}`,
+            `model = ${info.auto_model}.from_pretrained("${model.id}"` + remote_code_snippet + ")",
+        ].join("\n");
+    }
+    if (model.pipeline_tag && library_to_tasks_js_1.LIBRARY_TASK_MAPPING.transformers?.includes(model.pipeline_tag)) {
+        const pipelineSnippet = ["# Use a pipeline as a high-level helper", "from transformers import pipeline", ""];
+        if (model.tags.includes("conversational") && model.config?.tokenizer_config?.chat_template) {
+            pipelineSnippet.push("messages = [", '    {"role": "user", "content": "Who are you?"},', "]");
+        }
+        pipelineSnippet.push(`pipe = pipeline("${model.pipeline_tag}", model="${model.id}"` + remote_code_snippet + ")");
+        if (model.tags.includes("conversational") && model.config?.tokenizer_config?.chat_template) {
+            pipelineSnippet.push("pipe(messages)");
+        }
+        return [pipelineSnippet.join("\n"), autoSnippet];
+    }
+    return [autoSnippet];
+};
+exports.transformers = transformers;
+const transformersJS = (model) => {
+    if (!model.pipeline_tag) {
+        return [`// ⚠️ Unknown pipeline tag`];
+    }
+    const libName = "@huggingface/transformers";
+    return [
+        `// npm i ${libName}
+import { pipeline } from '${libName}';
+
+// Allocate pipeline
+const pipe = await pipeline('${model.pipeline_tag}', '${model.id}');`,
+    ];
+};
+exports.transformersJS = transformersJS;
+const peftTask = (peftTaskType) => {
+    switch (peftTaskType) {
+        case "CAUSAL_LM":
+            return "CausalLM";
+        case "SEQ_2_SEQ_LM":
+            return "Seq2SeqLM";
+        case "TOKEN_CLS":
+            return "TokenClassification";
+        case "SEQ_CLS":
+            return "SequenceClassification";
+        default:
+            return undefined;
+    }
+};
+const peft = (model) => {
+    const { base_model_name_or_path: peftBaseModel, task_type: peftTaskType } = model.config?.peft ?? {};
+    const pefttask = peftTask(peftTaskType);
+    if (!pefttask) {
+        return [`Task type is invalid.`];
+    }
+    if (!peftBaseModel) {
+        return [`Base model is not found.`];
+    }
+    return [
+        `from peft import PeftModel, PeftConfig
+from transformers import AutoModelFor${pefttask}
+
+config = PeftConfig.from_pretrained("${model.id}")
+base_model = AutoModelFor${pefttask}.from_pretrained("${peftBaseModel}")
+model = PeftModel.from_pretrained(base_model, "${model.id}")`,
+    ];
+};
+exports.peft = peft;
+const fasttext = (model) => [
+    `from huggingface_hub import hf_hub_download
+import fasttext
+
+model = fasttext.load_model(hf_hub_download("${model.id}", "model.bin"))`,
+];
+exports.fasttext = fasttext;
+const stableBaselines3 = (model) => [
+    `from huggingface_sb3 import load_from_hub
+checkpoint = load_from_hub(
+	repo_id="${model.id}",
+	filename="{MODEL FILENAME}.zip",
+)`,
+];
+exports.stableBaselines3 = stableBaselines3;
+const nemoDomainResolver = (domain, model) => {
+    switch (domain) {
+        case "ASR":
+            return [
+                `import nemo.collections.asr as nemo_asr
+asr_model = nemo_asr.models.ASRModel.from_pretrained("${model.id}")
+
+transcriptions = asr_model.transcribe(["file.wav"])`,
+            ];
+        default:
+            return undefined;
+    }
+};
+const mlAgents = (model) => [
+    `mlagents-load-from-hf --repo-id="${model.id}" --local-dir="./download: string[]s"`,
+];
+exports.mlAgents = mlAgents;
+const sentis = ( /* model: ModelData */) => [
+    `string modelName = "[Your model name here].sentis";
+Model model = ModelLoader.Load(Application.streamingAssetsPath + "/" + modelName);
+IWorker engine = WorkerFactory.CreateWorker(BackendType.GPUCompute, model);
+// Please see provided C# file for more details
+`,
+];
+exports.sentis = sentis;
+const vfimamba = (model) => [
+    `from Trainer_finetune import Model
+
+model = Model.from_pretrained("${model.id}")`,
+];
+exports.vfimamba = vfimamba;
+const voicecraft = (model) => [
+    `from voicecraft import VoiceCraft
+
+model = VoiceCraft.from_pretrained("${model.id}")`,
+];
+exports.voicecraft = voicecraft;
+const chattts = () => [
+    `import ChatTTS
+import torchaudio
+
+chat = ChatTTS.Chat()
+chat.load_models(compile=False) # Set to True for better performance
+
+texts = ["PUT YOUR TEXT HERE",]
+
+wavs = chat.infer(texts, )
+
+torchaudio.save("output1.wav", torch.from_numpy(wavs[0]), 24000)`,
+];
+exports.chattts = chattts;
+const yolov10 = (model) => [
+    `from ultralytics import YOLOv10
+
+model = YOLOv10.from_pretrained("${model.id}")
+source = 'http://images.cocodataset.org/val2017/000000039769.jpg'
+model.predict(source=source, save=True)
+`,
+];
+exports.yolov10 = yolov10;
+const birefnet = (model) => [
+    `# Option 1: use with transformers
+
+from transformers import AutoModelForImageSegmentation
+birefnet = AutoModelForImageSegmentation.from_pretrained("${model.id}", trust_remote_code=True)
+`,
+    `# Option 2: use with BiRefNet
+
+# Install from https://github.com/ZhengPeng7/BiRefNet
+
+from models.birefnet import BiRefNet
+model = BiRefNet.from_pretrained("${model.id}")`,
+];
+exports.birefnet = birefnet;
+const mlx = (model) => [
+    `pip install huggingface_hub hf_transfer
+
+export HF_HUB_ENABLE_HF_TRANS: string[]FER=1
+huggingface-cli download --local-dir ${nameWithoutNamespace(model.id)} ${model.id}`,
+];
+exports.mlx = mlx;
+const mlxim = (model) => [
+    `from mlxim.model import create_model
+
+model = create_model(${model.id})`,
+];
+exports.mlxim = mlxim;
+const model2vec = (model) => [
+    `from model2vec import StaticModel
+
+model = StaticModel.from_pretrained("${model.id}")`,
+];
+exports.model2vec = model2vec;
+const nemo = (model) => {
+    let command = undefined;
+    // Resolve the tag to a nemo domain/sub-domain
+    if (model.tags.includes("automatic-speech-recognition")) {
+        command = nemoDomainResolver("ASR", model);
+    }
+    return command ?? [`# tag did not correspond to a valid NeMo domain.`];
+};
+exports.nemo = nemo;
+const pxia = (model) => [
+    `from pxia import AutoModel
+
+model = AutoModel.from_pretrained("${model.id}")`,
+];
+exports.pxia = pxia;
+const pythae = (model) => [
+    `from pythae.models import AutoModel
+
+model = AutoModel.load_from_hf_hub("${model.id}")`,
+];
+exports.pythae = pythae;
+const musicgen = (model) => [
+    `from audiocraft.models import MusicGen
+
+model = MusicGen.get_pretrained("${model.id}")
+
+descriptions = ['happy rock', 'energetic EDM', 'sad jazz']
+wav = model.generate(descriptions)  # generates 3 samples.`,
+];
+const magnet = (model) => [
+    `from audiocraft.models import MAGNeT
+	
+model = MAGNeT.get_pretrained("${model.id}")
+
+descriptions = ['disco beat', 'energetic EDM', 'funky groove']
+wav = model.generate(descriptions)  # generates 3 samples.`,
+];
+const audiogen = (model) => [
+    `from audiocraft.models import AudioGen
+	
+model = AudioGen.get_pretrained("${model.id}")
+model.set_generation_params(duration=5)  # generate 5 seconds.
+descriptions = ['dog barking', 'sirene of an emergency vehicle', 'footsteps in a corridor']
+wav = model.generate(descriptions)  # generates 3 samples.`,
+];
+const audiocraft = (model) => {
+    if (model.tags.includes("musicgen")) {
+        return musicgen(model);
+    }
+    else if (model.tags.includes("audiogen")) {
+        return audiogen(model);
+    }
+    else if (model.tags.includes("magnet")) {
+        return magnet(model);
+    }
+    else {
+        return [`# Type of model unknown.`];
+    }
+};
+exports.audiocraft = audiocraft;
+const whisperkit = () => [
+    `# Install CLI with Homebrew on macOS device
+brew install whisperkit-cli
+
+# View all available inference options
+whisperkit-cli transcribe --help
+	
+# Download and run inference using whisper base model
+whisperkit-cli transcribe --audio-path /path/to/audio.mp3
+
+# Or use your preferred model variant
+whisperkit-cli transcribe --model "large-v3" --model-prefix "distil" --audio-path /path/to/audio.mp3 --verbose`,
+];
+exports.whisperkit = whisperkit;
+const threedtopia_xl = (model) => [
+    `from threedtopia_xl.models import threedtopia_xl
+
+model = threedtopia_xl.from_pretrained("${model.id}")
+model.generate(cond="path/to/image.png")`,
+];
+exports.threedtopia_xl = threedtopia_xl;
+const hezar = (model) => [
+    `from hezar import Model
+
+model = Model.load("${model.id}")`,
+];
+exports.hezar = hezar;
+//#endregion