llama-stack-mirror/llama_stack/templates
Ashwin Bharambe 04de2f84e9
fix: register provider model name and HF alias in run.yaml (#1304)
Each model known to the system has two identifiers: 

- the `provider_resource_id` (what the provider calls it) -- e.g.,
`accounts/fireworks/models/llama-v3p1-8b-instruct`
- the `identifier` (`model_id`) under which it is registered and gets
routed to the appropriate provider.

We have so far used the HuggingFace repo alias as the standardized
identifier you can use to refer to the model. So in the above example,
we'd use `meta-llama/Llama-3.1-8B-Instruct` as the name under which it
gets registered. This makes it convenient for users to refer to these
models across providers.

However, we forgot to register the _actual_ provider model ID also. You
should be able to route via `provider_resource_id` also, of course.

This change fixes this (somewhat grave) omission.

*Note*: this change is additive -- more aliases work now compared to
before.

## Test Plan

Run the following for distro=(ollama fireworks together)
```
LLAMA_STACK_CONFIG=$distro \
   pytest -s -v tests/client-sdk/inference/test_text_inference.py \
   --inference-model=meta-llama/Llama-3.1-8B-Instruct --vision-inference-model=""
```
2025-02-27 16:39:23 -08:00
..
bedrock fix: register provider model name and HF alias in run.yaml (#1304) 2025-02-27 16:39:23 -08:00
cerebras fix: register provider model name and HF alias in run.yaml (#1304) 2025-02-27 16:39:23 -08:00
ci-tests fix: register provider model name and HF alias in run.yaml (#1304) 2025-02-27 16:39:23 -08:00
dell fix: register provider model name and HF alias in run.yaml (#1304) 2025-02-27 16:39:23 -08:00
dev fix: register provider model name and HF alias in run.yaml (#1304) 2025-02-27 16:39:23 -08:00
experimental-post-training feat: [post training] support save hf safetensor format checkpoint (#845) 2025-02-25 23:29:08 -08:00
fireworks fix: register provider model name and HF alias in run.yaml (#1304) 2025-02-27 16:39:23 -08:00
groq fix: register provider model name and HF alias in run.yaml (#1304) 2025-02-27 16:39:23 -08:00
hf-endpoint fix: register provider model name and HF alias in run.yaml (#1304) 2025-02-27 16:39:23 -08:00
hf-serverless fix: register provider model name and HF alias in run.yaml (#1304) 2025-02-27 16:39:23 -08:00
meta-reference-gpu fix: register provider model name and HF alias in run.yaml (#1304) 2025-02-27 16:39:23 -08:00
meta-reference-quantized-gpu fix: register provider model name and HF alias in run.yaml (#1304) 2025-02-27 16:39:23 -08:00
nvidia fix: register provider model name and HF alias in run.yaml (#1304) 2025-02-27 16:39:23 -08:00
ollama fix: register provider model name and HF alias in run.yaml (#1304) 2025-02-27 16:39:23 -08:00
passthrough feat: inference passthrough provider (#1166) 2025-02-19 21:47:00 -08:00
remote-vllm fix: register provider model name and HF alias in run.yaml (#1304) 2025-02-27 16:39:23 -08:00
sambanova fix: register provider model name and HF alias in run.yaml (#1304) 2025-02-27 16:39:23 -08:00
tgi fix: register provider model name and HF alias in run.yaml (#1304) 2025-02-27 16:39:23 -08:00
together fix: register provider model name and HF alias in run.yaml (#1304) 2025-02-27 16:39:23 -08:00
vllm-gpu fix: register provider model name and HF alias in run.yaml (#1304) 2025-02-27 16:39:23 -08:00
__init__.py Auto-generate distro yamls + docs (#468) 2024-11-18 14:57:06 -08:00
template.py fix: register provider model name and HF alias in run.yaml (#1304) 2025-02-27 16:39:23 -08:00