mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-13 23:02:38 +00:00
dell tgi adapter
This commit is contained in:
parent
cfc97df6d5
commit
cd1f1a86bf
5 changed files with 61 additions and 11 deletions
49
tests/examples/dell-tgi-run.yaml
Normal file
49
tests/examples/dell-tgi-run.yaml
Normal file
|
|
@ -0,0 +1,49 @@
|
|||
version: '2'
|
||||
built_at: '2024-10-08T17:40:45.325529'
|
||||
image_name: local
|
||||
docker_image: null
|
||||
conda_env: local
|
||||
apis:
|
||||
- shields
|
||||
- agents
|
||||
- models
|
||||
- memory
|
||||
- memory_banks
|
||||
- inference
|
||||
- safety
|
||||
providers:
|
||||
inference:
|
||||
- provider_id: remote::dell-tgi
|
||||
provider_type: remote::dell-tgi
|
||||
config:
|
||||
url: http://127.0.0.1:5009
|
||||
hf_model_name: meta-llama/Llama-3.1-8B-Instruct
|
||||
safety:
|
||||
- provider_id: meta-reference
|
||||
provider_type: meta-reference
|
||||
config:
|
||||
llama_guard_shield:
|
||||
model: Llama-Guard-3-1B
|
||||
excluded_categories: []
|
||||
disable_input_check: false
|
||||
disable_output_check: false
|
||||
prompt_guard_shield:
|
||||
model: Prompt-Guard-86M
|
||||
memory:
|
||||
- provider_id: remote::chromadb
|
||||
provider_type: remote::chromadb
|
||||
config:
|
||||
host: localhost
|
||||
port: 6000
|
||||
agents:
|
||||
- provider_id: meta-reference
|
||||
provider_type: meta-reference
|
||||
config:
|
||||
persistence_store:
|
||||
namespace: null
|
||||
type: sqlite
|
||||
db_path: /home/xiyan/.llama/runtime/kvstore.db
|
||||
telemetry:
|
||||
- provider_id: meta-reference
|
||||
provider_type: meta-reference
|
||||
config: {}
|
||||
Loading…
Add table
Add a link
Reference in a new issue