diff --git a/README.md b/README.md index 3f75c495..c3cbc094 100644 --- a/README.md +++ b/README.md @@ -2,7 +2,7 @@ ## Setup your environment -In this directory, checkout the Foundation Model Stack (FMS) and the FMS Model Optimizer: +in this directory, checkout the Foundation Model Stack (FMS) and the FMS Model Optimizer: ```shell git clone https://github.com/foundation-model-stack/foundation-model-stack.git git clone https://github.com/foundation-model-stack/fms-model-optimizer.git diff --git a/tests/models/test_decoders.py b/tests/models/test_decoders.py index bb8a0e5b..165f775e 100644 --- a/tests/models/test_decoders.py +++ b/tests/models/test_decoders.py @@ -47,7 +47,7 @@ LLAMA_3p1_8B_INSTRUCT: os.path.join(MICRO_MODELS_HOME, "llama-3.1-8b-layers-3-step-24000"), 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.3-8b-layers-3-step-100000"), + GRANITE_3p3_8B_INSTRUCT: os.path.join(MICRO_MODELS_HOME, "granite-3.3-8b-layers-3-step-100000-bf16"), LLAMA_3p1_70B_INSTRUCT: os.path.join(MICRO_MODELS_HOME, "llama-3.1-70b-layers-3-step-24000") }