Skip to content

Commit cc180c2

Browse files
committed
Granite 3.3 ce and diff_mean values
Signed-off-by: Flavia Beo <flavia.beo@ibm.com>
1 parent bcc4c05 commit cc180c2

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

tests/models/test_decoders.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@
4040
# Add models to test here
4141
LLAMA_3p1_8B_INSTRUCT = "meta-llama/Llama-3.1-8B-Instruct"
4242
GRANITE_3p2_8B_INSTRUCT = "ibm-granite/granite-3.2-8b-instruct"
43+
GRANITE_3p3_8B_INSTRUCT = "ibm-granite/granite-3.3-8b-instruct"
4344
GRANITE_20B_CODE_INSTRUCT_8K = "ibm-granite/granite-20b-code-instruct-8k"
4445
LLAMA_3p1_70B_INSTRUCT = "meta-llama/Llama-3.1-70B-Instruct"
4546

@@ -65,6 +66,7 @@
6566
[
6667
LLAMA_3p1_8B_INSTRUCT,
6768
GRANITE_3p2_8B_INSTRUCT,
69+
GRANITE_3p3_8B_INSTRUCT,
6870
GRANITE_20B_CODE_INSTRUCT_8K,
6971
LLAMA_3p1_70B_INSTRUCT,
7072
],
@@ -147,6 +149,10 @@
147149
2.3919514417648315,
148150
0.0005767398688476533,
149151
),
152+
(GRANITE_3p3_8B_INSTRUCT, False): (
153+
2.4444521379470827,
154+
0.0004970188625156878,
155+
),
150156
(GRANITE_20B_CODE_INSTRUCT_8K, False): (
151157
2.640706129074097,
152158
0.00034344267623964697,

0 commit comments

Comments
 (0)