Skip to content

Commit 233c23f

Browse files
committed
removed one model by mistake
Signed-off-by: Yu Chin Fabian Lim <[email protected]>
1 parent 540594e commit 233c23f

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/models/test_decoders.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,8 @@
4848
GRANITE_3p2_8B_INSTRUCT: os.path.join(MICRO_MODELS_HOME, "granite-3.2-8b-layers-3-step-100000"),
4949
# FIXME: Because this uses the same config as 3.2, re-using here, but should update
5050
GRANITE_3p3_8B_INSTRUCT: os.path.join(MICRO_MODELS_HOME, "granite-3.2-8b-layers-3-step-100000"),
51-
GRANITE_20B_CODE_INSTRUCT_8K: os.path.join(MICRO_MODELS_HOME, "granite-20b-code-layers-3-step-100000")
51+
GRANITE_20B_CODE_INSTRUCT_8K: os.path.join(MICRO_MODELS_HOME, "granite-20b-code-layers-3-step-100000"),
52+
LLAMA_3p1_70B_INSTRUCT: os.path.join(MICRO_MODELS_HOME, "llama-3.1-70b-layers-3-step-24000")
5253
}
5354

5455
SHARE_GPT_DATASET_PATH = os.environ.get(

0 commit comments

Comments
 (0)