ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

TestNvidiaNim

Browse Source
This commit is contained in:
Ishaan Jaffer 2025-12-13 16:38:11 -08:00
parent f6c4ad92e4
commit accebc49a2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/llm_translation/test_nvidia_nim.py
View File

@ -253,7 +253,7 @@ class TestNvidiaNim(BaseLLMRerankTest):
def get_base_rerank_call_args(self) -> dict:
return {
"model": "nvidia_nim/nvidia/llama-3.2-nv-rerankqa-1b-v2",
"model": "nvidia_nim/nvidia/llama-3_2-nv-rerankqa-1b-v2",
}
def get_expected_cost(self) -> float: