diff --git a/tests/models/test_decoders.py b/tests/models/test_decoders.py index 45628a24..e89f9b94 100644 --- a/tests/models/test_decoders.py +++ b/tests/models/test_decoders.py @@ -56,7 +56,7 @@ ), # 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.3-8b-layers-3-step-100000" + MICRO_MODELS_HOME, "granite-3.3-8b-layers-4-step-100000-bf16" ), LLAMA_3p1_70B_INSTRUCT: os.path.join( MICRO_MODELS_HOME, "llama-3.1-70b-layers-3-step-24000"