[TestFix] change LoRA tests to use NVIDIA adapter instead of Nutanix (#19642)

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
Glen Liu
2026-03-02 15:55:41 -05:00
committed by GitHub
parent 51ee17ce44
commit cc860a2198
3 changed files with 60 additions and 81 deletions

View File

@@ -68,7 +68,7 @@ ALL_OTHER_LORA_MODELS = [
base="meta-llama/Llama-3.1-8B-Instruct",
adaptors=[
LoRAAdaptor(
name="Nutanix/Meta-Llama-3.1-8B-Instruct_lora_4_alpha_16",
name="nvidia/llama-3.1-nemoguard-8b-topic-control",
prefill_tolerance=1e-1,
),
],
@@ -109,7 +109,7 @@ ALL_OTHER_MULTI_LORA_MODELS = [
prefill_tolerance=1e-1,
),
LoRAAdaptor(
name="Nutanix/Meta-Llama-3.1-8B-Instruct_lora_4_alpha_16",
name="nvidia/llama-3.1-nemoguard-8b-topic-control",
prefill_tolerance=1e-1,
),
],