Skip to content

Commit 462122b

Browse files
committed
updated priority to frequency
Signed-off-by: Joshua Rosenkranz <jmrosenk@us.ibm.com>
1 parent c23ed14 commit 462122b

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

tests/models/test_decoders.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@
5656
[LLAMA_3p1_8B_INSTRUCT, GRANITE_3p2_8B_INSTRUCT, GRANITE_20B_CODE_INSTRUCT_8K, LLAMA_3p1_70B_INSTRUCT],
5757
)
5858
model_configuration_path = os.environ.get("FMS_TEST_SHAPES_FROM_MODEL_CONFIGURATION", "")
59-
model_configuration_priority = os.environ.get("FMS_TEST_SHAPES_FROM_MODEL_CONFIGURATION_PRIORITY", "0")
59+
model_configuration_frequency = os.environ.get("FMS_TEST_SHAPES_FROM_MODEL_CONFIGURATION_FREQUENCY", "0")
6060

6161
# for validation level 1, the default is a failure rate of 1%
6262
# set this environment variable if you would like to relax that threshold
@@ -145,12 +145,12 @@
145145
print("ignoring FMS_TEST_SHAPES_COMMON_MODEL_PATHS, FMS_TEST_SHAPES_USE_MICRO_MODELS as configuration will be set by FMS_TEST_SHAPES_FROM_MODEL_CONFIGURATION")
146146
USE_MICRO_MODELS = False
147147
common_model_paths = []
148-
priority = int(model_configuration_priority)
148+
frequency = int(model_configuration_frequency)
149149
with open(model_configuration_path, 'r') as f:
150150
for line in f:
151151
try:
152152
model_config = json.loads(line)
153-
if model_config["priority"] <= priority:
153+
if model_config["frequency"] <= frequency:
154154
common_model_paths.append(model_config["model_id"])
155155
# assume fullsize models
156156
fail_thresholds[(model_config["model_id"], USE_MICRO_MODELS)] = (model_config["ce"], model_config["mean_diff"])
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
1-
{"model_id": "mistralai/Mistral-7B-Instruct-v0.3", "ce": 2.8364005851745624, "mean_diff": 0.0007839603102183846, "priority": 2}
1+
{"model_id": "mistralai/Mistral-7B-Instruct-v0.3", "ce": 2.8364005851745624, "mean_diff": 0.0007839603102183846, "frequency": 2}
22
/* FIXME: proper mean_diff, currently using from granite 3.1 8b instruct */
3-
{"model_id": "ibm-granite/granite-guardian-3.1-8b", "ce": 2.493684446811673, "mean_diff": 0.0005767398688476533, "priority": 1}
3+
{"model_id": "ibm-granite/granite-guardian-3.1-8b", "ce": 2.493684446811673, "mean_diff": 0.0005767398688476533, "frequency": 1}

0 commit comments

Comments
 (0)