TGI tag only.

huggingface · Oct 9, 2024 · 933d60b · 933d60b
1 parent 9777175
commit 933d60b
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/packages/tasks/src/local-apps.ts b/packages/tasks/src/local-apps.ts
@@ -248,9 +248,7 @@ export const LOCAL_APPS = {
  prettyLabel: "TGI",
  docsUrl: "https://huggingface.co/docs/text-generation-inference/",
  mainTask: "text-generation",
- displayOnModelPage: (model: ModelData) =>
- (isAwqModel(model) || isGptqModel(model) || isMarlinModel(model) || isTransformersModel(model)) &&
- isTgiModel(model),
+ displayOnModelPage: (model: ModelData) => isTgiModel(model),
  snippet: snippetTgi,
  },
  lmstudio: {