@huggingface/tasks 0.19.83 → 0.19.85

package/src/local-apps.ts

@@ -91,6 +91,22 @@ function isLlamaCppGgufModel(model: ModelData) {
 	return !!model.gguf?.context_length;
 }
 
+function isVllmModel(model: ModelData): boolean {
+	return (
+		(isAwqModel(model) ||
+			isGptqModel(model) ||
+			isAqlmModel(model) ||
+			isMarlinModel(model) ||
+			isLlamaCppGgufModel(model) ||
+			isTransformersModel(model)) &&
+		(model.pipeline_tag === "text-generation" || model.pipeline_tag === "image-text-to-text")
+	);
+}
+
+function isDockerModelRunnerModel(model: ModelData): boolean {
+	return isLlamaCppGgufModel(model) || isVllmModel(model);
+}
+
 function isAmdRyzenModel(model: ModelData) {
 	return model.tags.includes("ryzenai-hybrid") || model.tags.includes("ryzenai-npu");
 }
@@ -152,6 +168,10 @@ const snippetLlamacpp = (model: ModelData, filepath?: string): LocalAppSnippet[]
 			].join("\n"),
 			content: [serverCommand("./build/bin/llama-server"), cliCommand("./build/bin/llama-cli")],
 		},
+		{
+			title: "Use Docker",
+			content: snippetDockerModelRunner(model, filepath),
+		},
 	];
 };
 
@@ -216,14 +236,6 @@ const snippetVllm = (model: ModelData): LocalAppSnippet[] => {
 	const serverCommand = `# Start the vLLM server:
 vllm serve "${model.id}"${mistralFlags}`;
 
-	const dockerCommand = `docker run --gpus all \\
-    -v ~/.cache/huggingface:/root/.cache/huggingface \\
-    --env "HF_TOKEN=<secret>" \\
-    -p 8000:8000 \\
-    --ipc=host \\
-    vllm/vllm-openai:latest \\
-    --model "${model.id}"${mistralFlags}`;
-
 	const runCommandInstruct = `# Call the server using curl (OpenAI-compatible API):
 curl -X POST "http://localhost:8000/v1/chat/completions" \\
 	-H "Content-Type: application/json" \\
@@ -253,9 +265,8 @@ curl -X POST "http://localhost:8000/v1/completions" \\
 			content: [serverCommand, runCommand],
 		},
 		{
-			title: "Use Docker images",
-			setup: dockerCommand,
-			content: [runCommand],
+			title: "Use Docker",
+			content: snippetDockerModelRunner(model),
 		},
 	];
 };
@@ -379,7 +390,9 @@ const snippetMlxLm = (model: ModelData): LocalAppSnippet[] => {
 };
 
 const snippetDockerModelRunner = (model: ModelData, filepath?: string): string => {
-	return `docker model run hf.co/${model.id}${getQuantTag(filepath)}`;
+	// Only add quant tag for GGUF models, not safetensors
+	const quantTag = isLlamaCppGgufModel(model) ? getQuantTag(filepath) : "";
+	return `docker model run hf.co/${model.id}${quantTag}`;
 };
 
 const snippetLemonade = (model: ModelData, filepath?: string): LocalAppSnippet[] => {
@@ -454,14 +467,7 @@ export const LOCAL_APPS = {
 		prettyLabel: "vLLM",
 		docsUrl: "https://docs.vllm.ai",
 		mainTask: "text-generation",
-		displayOnModelPage: (model: ModelData) =>
-			(isAwqModel(model) ||
-				isGptqModel(model) ||
-				isAqlmModel(model) ||
-				isMarlinModel(model) ||
-				isLlamaCppGgufModel(model) ||
-				isTransformersModel(model)) &&
-			(model.pipeline_tag === "text-generation" || model.pipeline_tag === "image-text-to-text"),
+		displayOnModelPage: isVllmModel,
 		snippet: snippetVllm,
 	},
 	sglang: {
@@ -604,7 +610,7 @@ export const LOCAL_APPS = {
 		prettyLabel: "Docker Model Runner",
 		docsUrl: "https://docs.docker.com/ai/model-runner/",
 		mainTask: "text-generation",
-		displayOnModelPage: isLlamaCppGgufModel,
+		displayOnModelPage: isDockerModelRunnerModel,
 		snippet: snippetDockerModelRunner,
 	},
 	lemonade: {

package/src/model-libraries-snippets.ts

@@ -1159,8 +1159,8 @@ for res in output:
 	}
 
 	if (model.tags.includes("document-parse")) {
-		const rawVersion = model.id.replace("PaddleOCR-VL-", "v");
-		const version = rawVersion === "PaddleOCR-VL" ? "v1" : rawVersion;
+		const rawVersion = model.id.replace("PaddlePaddle/PaddleOCR-VL-", "v");
+		const version = rawVersion === "PaddlePaddle/PaddleOCR-VL" ? "v1" : rawVersion;
 		return [
 			`# See https://www.paddleocr.ai/latest/version3.x/pipeline_usage/PaddleOCR-VL.html to installation
 
@@ -2277,6 +2277,29 @@ export const pythae = (model: ModelData): string[] => [
 model = AutoModel.load_from_hf_hub("${model.id}")`,
 ];
 
+export const qwen3_tts = (model: ModelData): string[] => [
+	`# pip install qwen-tts
+import torch
+import soundfile as sf
+from qwen_tts import Qwen3TTSModel
+
+model = Qwen3TTSModel.from_pretrained(
+    "${model.id}",
+    device_map="cuda:0",
+    dtype=torch.bfloat16,
+    attn_implementation="flash_attention_2",
+)
+
+wavs, sr = model.generate_custom_voice(
+    text="Your text here.",
+    language="English",
+    speaker="Ryan",
+    instruct="Speak in a natural tone.",
+)
+
+sf.write("output.wav", wavs[0], sr)`,
+];
+
 const musicgen = (model: ModelData): string[] => [
 	`from audiocraft.models import MusicGen
 

package/src/model-libraries.ts

@@ -629,6 +629,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 		filter: false,
 		countDownloads: `path:"infu_flux_v1.0/sim_stage1/image_proj_model.bin" OR path:"infu_flux_v1.0/aes_stage2/image_proj_model.bin"`,
 	},
+	intellifold: {
+		prettyLabel: "IntelliFold",
+		repoName: "IntelliFold",
+		repoUrl: "https://github.com/IntelliGen-AI/IntelliFold",
+		filter: false,
+		countDownloads: `path_extension:"pt"`,
+	},
 	keras: {
 		prettyLabel: "Keras",
 		repoName: "Keras",
@@ -1010,6 +1017,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 		filter: false,
 		countDownloads: `path_extension:"setup.py"`,
 	},
+	qwen3_tts: {
+		prettyLabel: "Qwen3-TTS",
+		repoName: "Qwen3-TTS",
+		repoUrl: "https://github.com/QwenLM/Qwen3-TTS",
+		snippets: snippets.qwen3_tts,
+		filter: false,
+	},
 	recurrentgemma: {
 		prettyLabel: "RecurrentGemma",
 		repoName: "recurrentgemma",
@@ -1418,6 +1432,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 		countDownloads: `path_extension:"pkl"`,
 		snippets: snippets.vfimamba,
 	},
+	vismatch: {
+		prettyLabel: "VisMatch",
+		repoName: "VisMatch",
+		repoUrl: "https://github.com/gmberton/vismatch",
+		filter: false,
+		countDownloads: `path:"vismatch.yaml"`,
+	},
 	lvface: {
 		prettyLabel: "LVFace",
 		repoName: "LVFace",