llama-stack-mirror/llama_stack/templates/tgi/build.yaml
Ashwin Bharambe 028530546f Convert TGI
2024-11-17 14:49:41 -08:00

19 lines
443 B
YAML

version: '2'
name: tgi
distribution_spec:
description: Use (an external) TGI server for running LLM inference
docker_image: llamastack/distribution-remote-tgi:test-0.0.52rc3
providers:
inference:
- remote::tgi
memory:
- inline::faiss
- remote::chromadb
- remote::pgvector
safety:
- inline::llama-guard
agents:
- inline::meta-reference
telemetry:
- inline::meta-reference
image_type: conda