diff --git a/tests/models/test_decoders.py b/tests/models/test_decoders.py index ad4bcb97..e8fbd58b 100644 --- a/tests/models/test_decoders.py +++ b/tests/models/test_decoders.py @@ -48,6 +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"), + 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") }