phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2026-01-02 07:44:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix intergration tests

Browse source
This commit is contained in:
Honglin Cao 2025-03-11 11:18:50 -04:00
parent d1f67d90ca
commit a454b53bda
5 changed files with 9 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/integration/inference/test_text_inference.py
View file

@ -12,7 +12,7 @@ from llama_stack.models.llama.sku_list import resolve_model
from ..test_cases.test_case import TestCase
PROVIDER_LOGPROBS_TOP_K = {"remote::together", "remote::fireworks", "remote::vllm"}
PROVIDER_LOGPROBS_TOP_K = {"remote::together", "remote::fireworks", "remote::vllm", "remote::centml"}
def skip_if_model_doesnt_support_completion(client_with_models, model_id):