TestNvidiaNim

This commit is contained in:
Ishaan Jaffer
2025-12-13 16:38:11 -08:00
parent f6c4ad92e4
commit accebc49a2
+1 -1
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: