Update packages/tasks/src/local-apps.ts

huggingface · Oct 9, 2024 · 66745b7 · 66745b7
1 parent 933d60b
commit 66745b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/tasks/src/local-apps.ts b/packages/tasks/src/local-apps.ts
@@ -203,7 +203,7 @@ const snippetTgi = (model: ModelData): LocalAppSnippet[] => {
  `docker run --gpus all \\`,
  ` -v ~/.cache/huggingface:/root/.cache/huggingface \\`,
  ` -e HF_TOKEN=<secret>" \\`,
- ` ghcr.io/huggingface/text-generation-inference:2.3.1 \\`,
+ ` ghcr.io/huggingface/text-generation-inference:latest \\`,
  ` --model-id ${model.id}`,
  ].join("\n"),
  content: [runCommand.join("\n")],