djstrong commited on
Commit
8048010
1 Parent(s): 0a713c8
Files changed (1) hide show
  1. src/leaderboard/read_evals.py +4 -0
src/leaderboard/read_evals.py CHANGED
@@ -63,6 +63,7 @@ class EvalResult:
63
 
64
  org_and_model = org_and_model.replace(",dtype=bfloat16", "")
65
  org_and_model = org_and_model.replace(",dtype=float16", "")
 
66
 
67
  org_and_model = org_and_model.replace("models/hf_v7_e1", "APT3-1B-Instruct-e1")
68
  org_and_model = org_and_model.replace("models/hf_v7_e2", "APT3-1B-Instruct-e2")
@@ -79,6 +80,9 @@ class EvalResult:
79
  org_and_model = re.sub(",prefix_token_id=\d+", "", org_and_model)
80
  org_and_model = re.sub("/$", "", org_and_model)
81
 
 
 
 
82
  model_mapping={
83
  'speakleash/mistral_7B-v2/spkl-only-e1_333887a5':'speakleash/Bielik-7B-v0.1',
84
  'speakleash/mistral_7B-v2/spkl-only_sft_v2/e1_base/spkl-only_v10wa_7e6-e2_bbc67e89':'speakleash/Bielik-7B-Instruct-v0.1',
 
63
 
64
  org_and_model = org_and_model.replace(",dtype=bfloat16", "")
65
  org_and_model = org_and_model.replace(",dtype=float16", "")
66
+ org_and_model = org_and_model.replace(",tokenizer=.*", "")
67
 
68
  org_and_model = org_and_model.replace("models/hf_v7_e1", "APT3-1B-Instruct-e1")
69
  org_and_model = org_and_model.replace("models/hf_v7_e2", "APT3-1B-Instruct-e2")
 
80
  org_and_model = re.sub(",prefix_token_id=\d+", "", org_and_model)
81
  org_and_model = re.sub("/$", "", org_and_model)
82
 
83
+ if 'gguf-iq' in org_and_model:
84
+ org_and_model = org_and_model + '.IQ'
85
+
86
  model_mapping={
87
  'speakleash/mistral_7B-v2/spkl-only-e1_333887a5':'speakleash/Bielik-7B-v0.1',
88
  'speakleash/mistral_7B-v2/spkl-only_sft_v2/e1_base/spkl-only_v10wa_7e6-e2_bbc67e89':'speakleash/Bielik-7B-Instruct-v0.1',