From 801b0c34427813a191a6952fd41259f6793e1541 Mon Sep 17 00:00:00 2001 From: Yu Chin Fabian Lim Date: Wed, 25 Jun 2025 00:04:10 +0000 Subject: [PATCH 1/2] 20b tiny model Signed-off-by: Yu Chin Fabian Lim --- tests/models/test_decoders.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/test_decoders.py b/tests/models/test_decoders.py index ad4bcb97..1bfe259f 100644 --- a/tests/models/test_decoders.py +++ b/tests/models/test_decoders.py @@ -48,7 +48,7 @@ GRANITE_3p2_8B_INSTRUCT: os.path.join(MICRO_MODELS_HOME, "granite-3.2-8b-layers-3-step-100000"), # FIXME: Because this uses the same config as 3.2, re-using here, but should update GRANITE_3p3_8B_INSTRUCT: os.path.join(MICRO_MODELS_HOME, "granite-3.2-8b-layers-3-step-100000"), - LLAMA_3p1_70B_INSTRUCT: os.path.join(MICRO_MODELS_HOME, "llama-3.1-70b-layers-3-step-24000") + GRANITE_20B_CODE_INSTRUCT_8K: os.path.join(MICRO_MODELS_HOME, "granite-20b-code-layers-3-step-100000") } SHARE_GPT_DATASET_PATH = os.environ.get( From 4ed8b954180dc8f6b01a1f4f208ad9eef1928a00 Mon Sep 17 00:00:00 2001 From: Yu Chin Fabian Lim Date: Wed, 25 Jun 2025 01:11:37 +0000 Subject: [PATCH 2/2] removed one model by mistake Signed-off-by: Yu Chin Fabian Lim --- tests/models/test_decoders.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/models/test_decoders.py b/tests/models/test_decoders.py index 1bfe259f..e8fbd58b 100644 --- a/tests/models/test_decoders.py +++ b/tests/models/test_decoders.py @@ -48,7 +48,8 @@ GRANITE_3p2_8B_INSTRUCT: os.path.join(MICRO_MODELS_HOME, "granite-3.2-8b-layers-3-step-100000"), # FIXME: Because this uses the same config as 3.2, re-using here, but should update GRANITE_3p3_8B_INSTRUCT: os.path.join(MICRO_MODELS_HOME, "granite-3.2-8b-layers-3-step-100000"), - GRANITE_20B_CODE_INSTRUCT_8K: os.path.join(MICRO_MODELS_HOME, "granite-20b-code-layers-3-step-100000") + GRANITE_20B_CODE_INSTRUCT_8K: os.path.join(MICRO_MODELS_HOME, "granite-20b-code-layers-3-step-100000"), + LLAMA_3p1_70B_INSTRUCT: os.path.join(MICRO_MODELS_HOME, "llama-3.1-70b-layers-3-step-24000") } SHARE_GPT_DATASET_PATH = os.environ.get(