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

Signed-off-by: Wallas Santos <wallashss@ibm.com>
2024-10-22 01:52:14 -03:00
parent 74692421f7
commit c0292211ce
8 changed files with 15 additions and 15 deletions
--- a/tests/basic_correctness/test_cpu_offload.py
+++ b/tests/basic_correctness/test_cpu_offload.py
@@ -2,5 +2,5 @@ from ..utils import compare_two_settings


 def test_cpu_offload():
-    compare_two_settings("meta-llama/Llama-2-7b-hf", [],
-                         ["--cpu-offload-gb", "4"])
+    compare_two_settings("meta-llama/Llama-3.2-1B", [],
+                         ["--cpu-offload-gb", "1"])