[vLLM Snippet] pipeline has to be one of "text-generation", "image-te…

…xt-to-text"
huggingface · Oct 7, 2024 · bd1c0a2 · bd1c0a2
1 parent e60e407
commit bd1c0a2
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/packages/tasks/src/local-apps.ts b/packages/tasks/src/local-apps.ts
@@ -205,12 +205,13 @@ export const LOCAL_APPS = {
 		docsUrl: "https://docs.vllm.ai",
 		mainTask: "text-generation",
 		displayOnModelPage: (model: ModelData) =>
-			isAwqModel(model) ||
-			isGptqModel(model) ||
-			isAqlmModel(model) ||
-			isMarlinModel(model) ||
-			isLlamaCppGgufModel(model) ||
-			isTransformersModel(model),
+			(isAwqModel(model) ||
+				isGptqModel(model) ||
+				isAqlmModel(model) ||
+				isMarlinModel(model) ||
+				isLlamaCppGgufModel(model) ||
+				isTransformersModel(model)) &&
+			(model.pipeline_tag === "text-generation" || model.pipeline_tag === "image-text-to-text"),
 		snippet: snippetVllm,
 	},
 	lmstudio: {