[CI/Build] Replaced some models on tests for smaller ones (#9570)

Signed-off-by: Wallas Santos <wallashss@ibm.com>
This commit is contained in:
Wallas Henrique
2024-10-22 01:52:14 -03:00
committed by GitHub
parent 74692421f7
commit c0292211ce
8 changed files with 15 additions and 15 deletions

View File

@@ -16,7 +16,7 @@ from ..utils import multi_gpu_test
MODELS = [
"facebook/opt-125m",
"meta-llama/Llama-2-7b-hf",
"meta-llama/Llama-3.2-1B",
]