ClΓ©mentine commited on
Commit
9d6aecc
β€’
1 Parent(s): 9d8281c

update information base

Browse files
app.py CHANGED
@@ -318,7 +318,7 @@ with demo:
318
  queue=True,
319
  )
320
 
321
- with gr.TabItem("πŸ“ˆ Metrics through time", elem_id="llm-benchmark-tab-table", id=4):
322
  with gr.Row():
323
  with gr.Column():
324
  chart = create_metric_plot_obj(
@@ -334,50 +334,17 @@ with demo:
334
  title="Top Scores and Human Baseline Over Time (from last update)",
335
  )
336
  gr.Plot(value=chart, min_width=500)
337
- with gr.TabItem("πŸ“ About", elem_id="llm-benchmark-tab-table", id=2):
338
  gr.Markdown(LLM_BENCHMARKS_TEXT, elem_classes="markdown-text")
 
 
339
  gr.Markdown(FAQ_TEXT, elem_classes="markdown-text")
340
 
341
- with gr.TabItem("πŸš€ Submit here! ", elem_id="llm-benchmark-tab-table", id=3):
342
  with gr.Column():
343
  with gr.Row():
344
  gr.Markdown(EVALUATION_QUEUE_TEXT, elem_classes="markdown-text")
345
 
346
- with gr.Column():
347
- with gr.Accordion(
348
- f"βœ… Finished Evaluations ({len(finished_eval_queue_df)})",
349
- open=False,
350
- ):
351
- with gr.Row():
352
- finished_eval_table = gr.components.Dataframe(
353
- value=finished_eval_queue_df,
354
- headers=EVAL_COLS,
355
- datatype=EVAL_TYPES,
356
- row_count=5,
357
- )
358
- with gr.Accordion(
359
- f"πŸ”„ Running Evaluation Queue ({len(running_eval_queue_df)})",
360
- open=False,
361
- ):
362
- with gr.Row():
363
- running_eval_table = gr.components.Dataframe(
364
- value=running_eval_queue_df,
365
- headers=EVAL_COLS,
366
- datatype=EVAL_TYPES,
367
- row_count=5,
368
- )
369
-
370
- with gr.Accordion(
371
- f"⏳ Pending Evaluation Queue ({len(pending_eval_queue_df)})",
372
- open=False,
373
- ):
374
- with gr.Row():
375
- pending_eval_table = gr.components.Dataframe(
376
- value=pending_eval_queue_df,
377
- headers=EVAL_COLS,
378
- datatype=EVAL_TYPES,
379
- row_count=5,
380
- )
381
  with gr.Row():
382
  gr.Markdown("# βœ‰οΈβœ¨ Submit your model here!", elem_classes="markdown-text")
383
 
@@ -411,6 +378,42 @@ with demo:
411
  )
412
  base_model_name_textbox = gr.Textbox(label="Base model (for delta or adapter weights)")
413
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
414
  submit_button = gr.Button("Submit Eval")
415
  submission_result = gr.Markdown()
416
  submit_button.click(
 
318
  queue=True,
319
  )
320
 
321
+ with gr.TabItem("πŸ“ˆ Metrics through time", elem_id="llm-benchmark-tab-table", id=2):
322
  with gr.Row():
323
  with gr.Column():
324
  chart = create_metric_plot_obj(
 
334
  title="Top Scores and Human Baseline Over Time (from last update)",
335
  )
336
  gr.Plot(value=chart, min_width=500)
337
+ with gr.TabItem("πŸ“ About", elem_id="llm-benchmark-tab-table", id=3):
338
  gr.Markdown(LLM_BENCHMARKS_TEXT, elem_classes="markdown-text")
339
+
340
+ with gr.TabItem("❗FAQ", elem_id="llm-benchmark-tab-table", id=4):
341
  gr.Markdown(FAQ_TEXT, elem_classes="markdown-text")
342
 
343
+ with gr.TabItem("πŸš€ Submit here! ", elem_id="llm-benchmark-tab-table", id=5):
344
  with gr.Column():
345
  with gr.Row():
346
  gr.Markdown(EVALUATION_QUEUE_TEXT, elem_classes="markdown-text")
347
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
348
  with gr.Row():
349
  gr.Markdown("# βœ‰οΈβœ¨ Submit your model here!", elem_classes="markdown-text")
350
 
 
378
  )
379
  base_model_name_textbox = gr.Textbox(label="Base model (for delta or adapter weights)")
380
 
381
+ with gr.Column():
382
+ with gr.Accordion(
383
+ f"βœ… Finished Evaluations ({len(finished_eval_queue_df)})",
384
+ open=False,
385
+ ):
386
+ with gr.Row():
387
+ finished_eval_table = gr.components.Dataframe(
388
+ value=finished_eval_queue_df,
389
+ headers=EVAL_COLS,
390
+ datatype=EVAL_TYPES,
391
+ row_count=5,
392
+ )
393
+ with gr.Accordion(
394
+ f"πŸ”„ Running Evaluation Queue ({len(running_eval_queue_df)})",
395
+ open=False,
396
+ ):
397
+ with gr.Row():
398
+ running_eval_table = gr.components.Dataframe(
399
+ value=running_eval_queue_df,
400
+ headers=EVAL_COLS,
401
+ datatype=EVAL_TYPES,
402
+ row_count=5,
403
+ )
404
+
405
+ with gr.Accordion(
406
+ f"⏳ Pending Evaluation Queue ({len(pending_eval_queue_df)})",
407
+ open=False,
408
+ ):
409
+ with gr.Row():
410
+ pending_eval_table = gr.components.Dataframe(
411
+ value=pending_eval_queue_df,
412
+ headers=EVAL_COLS,
413
+ datatype=EVAL_TYPES,
414
+ row_count=5,
415
+ )
416
+
417
  submit_button = gr.Button("Submit Eval")
418
  submission_result = gr.Markdown()
419
  submit_button.click(
src/display/about.py CHANGED
@@ -13,13 +13,19 @@ Other cool leaderboards:
13
  - [LLM performance](https://huggingface.co/spaces/optimum/llm-perf-leaderboard)
14
  """
15
 
 
 
 
 
 
 
 
16
  LLM_BENCHMARKS_TEXT = f"""
17
- # Context
18
  With the plethora of large language models (LLMs) and chatbots being released week upon week, often with grandiose claims of their performance, it can be hard to filter out the genuine progress that is being made by the open-source community and which model is the current state of the art.
19
 
20
- ## How it works
21
-
22
- πŸ“ˆ We evaluate models on 7 key benchmarks using the <a href="https://github.com/EleutherAI/lm-evaluation-harness" target="_blank"> Eleuther AI Language Model Evaluation Harness </a>, a unified framework to test generative language models on a large number of different evaluation tasks.
23
 
24
  - <a href="https://arxiv.org/abs/1803.05457" target="_blank"> AI2 Reasoning Challenge </a> (25-shot) - a set of grade-school science questions.
25
  - <a href="https://arxiv.org/abs/1905.07830" target="_blank"> HellaSwag </a> (10-shot) - a test of commonsense inference, which is easy for humans (~95%) but challenging for SOTA models.
@@ -31,14 +37,20 @@ With the plethora of large language models (LLMs) and chatbots being released we
31
  For all these evaluations, a higher score is a better score.
32
  We chose these benchmarks as they test a variety of reasoning and general knowledge across a wide variety of fields in 0-shot and few-shot settings.
33
 
34
- ## Details and logs
35
  You can find:
36
  - detailed numerical results in the `results` Hugging Face dataset: https://huggingface.co/datasets/open-llm-leaderboard/results
37
  - details on the input/outputs for the models in the `details` of each model, that you can access by clicking the πŸ“„ emoji after the model name
38
  - community queries and running status in the `requests` Hugging Face dataset: https://huggingface.co/datasets/open-llm-leaderboard/requests
39
 
40
- ## Reproducibility
41
- To reproduce our results, use [this version](https://github.com/EleutherAI/lm-evaluation-harness/tree/b281b0921b636bc36ad05c0b0b0763bd6dd43463) of the Eleuther AI Harness and run:
 
 
 
 
 
 
42
 
43
  ```
44
  python main.py --model=hf-causal-experimental \
@@ -64,31 +76,23 @@ Side note on the baseline scores:
64
  - for log-likelihood evaluation, we select the random baseline
65
  - for GSM8K, we select the score obtained in the paper after finetuning a 6B model on the full GSM8K training set for 50 epochs
66
 
67
- ## Icons
68
- - {ModelType.PT.to_str(" : ")} model: new, base models, trained on a given corpora
69
- - {ModelType.FT.to_str(" : ")} model: pretrained models finetuned on more data
70
- - {ModelType.chat.to_str(" : ")} model: chat like fine-tunes, either using IFT (datasets of task instruction), RLHF or DPO (changing the model loss a bit with an added policy), etc
71
- - {ModelType.merges.to_str(" : ")} model: merges or MoErges, models which have been merged or fused without additional fine-tuning.
72
- If there is no icon, we have not uploaded the information on the model yet, feel free to open an issue with the model information!
73
 
74
- "Flagged" indicates that this model has been flagged by the community, and should probably be ignored! Clicking the link will redirect you to the discussion about the model.
75
 
76
- ## Quantization
77
  To get more information about quantization, see:
78
  - 8 bits: [blog post](https://huggingface.co/blog/hf-bitsandbytes-integration), [paper](https://arxiv.org/abs/2208.07339)
79
  - 4 bits: [blog post](https://huggingface.co/blog/4bit-transformers-bitsandbytes), [paper](https://arxiv.org/abs/2305.14314)
80
 
81
- ## Useful links
82
  - [Community resources](https://huggingface.co/spaces/HuggingFaceH4/open_llm_leaderboard/discussions/174)
83
  - [Collection of best models](https://huggingface.co/collections/open-llm-leaderboard/llm-leaderboard-best-models-652d6c7965a4619fb5c27a03)
84
  """
85
 
86
  FAQ_TEXT = """
87
- ---------------------------
88
- # FAQ
89
- Below are some common questions - if this FAQ does not answer you, feel free to create a new issue, and we'll take care of it as soon as we can!
90
 
91
- ## 1) Submitting a model
92
  My model requires `trust_remote_code=True`, can I submit it?
93
  - *We only support models that have been integrated in a stable version of the `transformers` library for automatic submission, as we don't want to run possibly unsage code on our cluster.*
94
 
@@ -108,7 +112,9 @@ How can I report an evaluation failure?
108
  - *As we store the logs for all models, feel free to create an issue, **where you link to the requests file of your model** (look for it [here](https://huggingface.co/datasets/open-llm-leaderboard/requests/tree/main)), so we can investigate! If the model failed due to a problem on our side, we'll relaunch it right away!*
109
  *Note: Please do not re-upload your model under a different name, it will not help*
110
 
111
- ## 2) Model results
 
 
112
  What kind of information can I find?
113
  - *Let's imagine you are interested in the Yi-34B results. You have access to 3 different information categories:*
114
  - *The [request file](https://huggingface.co/datasets/open-llm-leaderboard/requests/blob/main/01-ai/Yi-34B_eval_request_False_bfloat16_Original.json): it gives you information about the status of the evaluation*
@@ -125,27 +131,44 @@ What is this concept of "flagging"?
125
  My model has been flagged improperly, what can I do?
126
  - *Every flagged model has a discussion associated with it - feel free to plead your case there, and we'll see what to do together with the community.*
127
 
128
- ## 3) Editing a submission
 
 
129
  I upgraded my model and want to re-submit, how can I do that?
130
  - *Please open an issue with the precise name of your model, and we'll remove your model from the leaderboard so you can resubmit. You can also resubmit directly with the new commit hash!*
131
 
132
  I need to rename my model, how can I do that?
133
  - *You can use @Weyaxi 's [super cool tool](https://huggingface.co/spaces/Weyaxi/open-llm-leaderboard-renamer) to request model name changes, then open a discussion where you link to the created pull request, and we'll check them and merge them as needed.*
134
 
135
- ## 4) Other
 
 
 
 
 
 
 
 
 
 
136
  Why don't you display closed source model scores?
137
  - *This is a leaderboard for Open models, both for philosophical reasons (openness is cool) and for practical reasons: we want to ensure that the results we display are accurate and reproducible, but 1) commercial closed models can change their API thus rendering any scoring at a given time incorrect 2) we re-run everything on our cluster to ensure all models are run on the same setup and you can't do that for these models.*
138
 
139
  I have an issue about accessing the leaderboard through the Gradio API
140
  - *Since this is not the recommended way to access the leaderboard, we won't provide support for this, but you can look at tools provided by the community for inspiration!*
 
 
 
141
  """
142
 
143
 
144
- EVALUATION_QUEUE_TEXT = """
145
  # Evaluation Queue for the πŸ€— Open LLM Leaderboard
146
 
147
  Models added here will be automatically evaluated on the πŸ€— cluster.
148
 
 
 
149
  ## First steps before submitting a model
150
 
151
  ### 1) Make sure you can load your model and tokenizer using AutoClasses:
@@ -172,10 +195,8 @@ When we add extra information about models to the leaderboard, it will be automa
172
  ### 5) Select the correct precision
173
  Not all models are converted properly from `float16` to `bfloat16`, and selecting the wrong precision can sometimes cause evaluation error (as loading a `bf16` model in `fp16` can sometimes generate NaNs, depending on the weight range).
174
 
175
- ## In case of model failure
176
- If your model is displayed in the `FAILED` category, its execution stopped.
177
- Make sure you have followed the above steps first.
178
- If everything is done, check you can launch the EleutherAIHarness on your model locally, using the command in the About tab under "Reproducibility" with all arguments specified (you can add `--limit` to limit the number of examples per task).
179
  """
180
 
181
  CITATION_BUTTON_LABEL = "Copy the following snippet to cite these results"
 
13
  - [LLM performance](https://huggingface.co/spaces/optimum/llm-perf-leaderboard)
14
  """
15
 
16
+ icons = f"""
17
+ - {ModelType.PT.to_str(" : ")} model: new, base models, trained on a given corpora
18
+ - {ModelType.CPT.to_str(" : ")} model: new, base models, continuously trained on a given corpora, which includes IFT/chat data
19
+ - {ModelType.FT.to_str(" : ")} model: pretrained models finetuned on more data
20
+ - {ModelType.chat.to_str(" : ")} model: chat like fine-tunes, either using IFT (datasets of task instruction), RLHF or DPO (changing the model loss a bit with an added policy), etc
21
+ - {ModelType.merges.to_str(" : ")} model: merges or MoErges, models which have been merged or fused without additional fine-tuning.
22
+ """
23
  LLM_BENCHMARKS_TEXT = f"""
24
+ ## ABOUT
25
  With the plethora of large language models (LLMs) and chatbots being released week upon week, often with grandiose claims of their performance, it can be hard to filter out the genuine progress that is being made by the open-source community and which model is the current state of the art.
26
 
27
+ ### Tasks
28
+ πŸ“ˆ We evaluate models on 6 key benchmarks using the <a href="https://github.com/EleutherAI/lm-evaluation-harness" target="_blank"> Eleuther AI Language Model Evaluation Harness </a>, a unified framework to test generative language models on a large number of different evaluation tasks.
 
29
 
30
  - <a href="https://arxiv.org/abs/1803.05457" target="_blank"> AI2 Reasoning Challenge </a> (25-shot) - a set of grade-school science questions.
31
  - <a href="https://arxiv.org/abs/1905.07830" target="_blank"> HellaSwag </a> (10-shot) - a test of commonsense inference, which is easy for humans (~95%) but challenging for SOTA models.
 
37
  For all these evaluations, a higher score is a better score.
38
  We chose these benchmarks as they test a variety of reasoning and general knowledge across a wide variety of fields in 0-shot and few-shot settings.
39
 
40
+ ### Results
41
  You can find:
42
  - detailed numerical results in the `results` Hugging Face dataset: https://huggingface.co/datasets/open-llm-leaderboard/results
43
  - details on the input/outputs for the models in the `details` of each model, that you can access by clicking the πŸ“„ emoji after the model name
44
  - community queries and running status in the `requests` Hugging Face dataset: https://huggingface.co/datasets/open-llm-leaderboard/requests
45
 
46
+ If a model's name contains "Flagged", this indicates it has been flagged by the community, and should probably be ignored! Clicking the link will redirect you to the discussion about the model.
47
+
48
+ ---------------------------
49
+
50
+ ## REPRODUCIBILITY
51
+ To reproduce our results, here is the commands you can run, using [this version](https://github.com/EleutherAI/lm-evaluation-harness/tree/b281b0921b636bc36ad05c0b0b0763bd6dd43463) of the Eleuther AI Harness:
52
+ `python main.py --model=hf-causal-experimental --model_args="pretrained=<your_model>,use_accelerate=True,revision=<your_model_revision>"`
53
+ ` --tasks=<task_list> --num_fewshot=<n_few_shot> --batch_size=1 --output_path=<output_path>`
54
 
55
  ```
56
  python main.py --model=hf-causal-experimental \
 
76
  - for log-likelihood evaluation, we select the random baseline
77
  - for GSM8K, we select the score obtained in the paper after finetuning a 6B model on the full GSM8K training set for 50 epochs
78
 
79
+ ---------------------------
 
 
 
 
 
80
 
81
+ ## RESSOURCES
82
 
83
+ ### Quantization
84
  To get more information about quantization, see:
85
  - 8 bits: [blog post](https://huggingface.co/blog/hf-bitsandbytes-integration), [paper](https://arxiv.org/abs/2208.07339)
86
  - 4 bits: [blog post](https://huggingface.co/blog/4bit-transformers-bitsandbytes), [paper](https://arxiv.org/abs/2305.14314)
87
 
88
+ ### Useful links
89
  - [Community resources](https://huggingface.co/spaces/HuggingFaceH4/open_llm_leaderboard/discussions/174)
90
  - [Collection of best models](https://huggingface.co/collections/open-llm-leaderboard/llm-leaderboard-best-models-652d6c7965a4619fb5c27a03)
91
  """
92
 
93
  FAQ_TEXT = """
 
 
 
94
 
95
+ ## SUBMISSIONS
96
  My model requires `trust_remote_code=True`, can I submit it?
97
  - *We only support models that have been integrated in a stable version of the `transformers` library for automatic submission, as we don't want to run possibly unsage code on our cluster.*
98
 
 
112
  - *As we store the logs for all models, feel free to create an issue, **where you link to the requests file of your model** (look for it [here](https://huggingface.co/datasets/open-llm-leaderboard/requests/tree/main)), so we can investigate! If the model failed due to a problem on our side, we'll relaunch it right away!*
113
  *Note: Please do not re-upload your model under a different name, it will not help*
114
 
115
+ ---------------------------
116
+
117
+ ## RESULTS
118
  What kind of information can I find?
119
  - *Let's imagine you are interested in the Yi-34B results. You have access to 3 different information categories:*
120
  - *The [request file](https://huggingface.co/datasets/open-llm-leaderboard/requests/blob/main/01-ai/Yi-34B_eval_request_False_bfloat16_Original.json): it gives you information about the status of the evaluation*
 
131
  My model has been flagged improperly, what can I do?
132
  - *Every flagged model has a discussion associated with it - feel free to plead your case there, and we'll see what to do together with the community.*
133
 
134
+ ---------------------------
135
+
136
+ ## EDITING SUBMISSIONS
137
  I upgraded my model and want to re-submit, how can I do that?
138
  - *Please open an issue with the precise name of your model, and we'll remove your model from the leaderboard so you can resubmit. You can also resubmit directly with the new commit hash!*
139
 
140
  I need to rename my model, how can I do that?
141
  - *You can use @Weyaxi 's [super cool tool](https://huggingface.co/spaces/Weyaxi/open-llm-leaderboard-renamer) to request model name changes, then open a discussion where you link to the created pull request, and we'll check them and merge them as needed.*
142
 
143
+ ---------------------------
144
+
145
+ ## OTHER
146
+ Why do you differentiate between pretrained, continously pretrained, fine-tuned, merges, etc ?
147
+ - *These different models do not play in the same categories, and therefore need to be separated for fair comparision. Base pretrained models are the most interesting for the community, as they are usually good models to fine-tune later on - any jump in performance from a pretrained model represents a true improvement on the SOTA.
148
+ Fine tuned and IFT/RLHF/chat models usually have better performance, but the latter might be more sensitive to system prompts, which we do not cover at the moment in the Open LLM Leaderboard.
149
+ Merges and moerges have artificially inflated performance on test sets, which is not always explainable, and does not always apply to real world situations.*
150
+
151
+ What should I use the leaderboard for?
152
+ - *We recommend using the leaderboard for 3 use cases: 1) getting an idea of the state of open pretrained models, by looking only at the ranks and score of this category; 2) experimenting with different fine tuning methods, datasets, quantization techniques, etc, and comparing their score in a reproducible setup, and 3) checking the performance of a model of interest to you, wrt to other models of its category.*
153
+
154
  Why don't you display closed source model scores?
155
  - *This is a leaderboard for Open models, both for philosophical reasons (openness is cool) and for practical reasons: we want to ensure that the results we display are accurate and reproducible, but 1) commercial closed models can change their API thus rendering any scoring at a given time incorrect 2) we re-run everything on our cluster to ensure all models are run on the same setup and you can't do that for these models.*
156
 
157
  I have an issue about accessing the leaderboard through the Gradio API
158
  - *Since this is not the recommended way to access the leaderboard, we won't provide support for this, but you can look at tools provided by the community for inspiration!*
159
+
160
+ I have another problem, help!
161
+ - *Please open an issue in the discussion tab, and we'll do our best to help you in a timely manner :) *
162
  """
163
 
164
 
165
+ EVALUATION_QUEUE_TEXT = f"""
166
  # Evaluation Queue for the πŸ€— Open LLM Leaderboard
167
 
168
  Models added here will be automatically evaluated on the πŸ€— cluster.
169
 
170
+ ## Don't forget to read the FAQ and the About tabs for more information!
171
+
172
  ## First steps before submitting a model
173
 
174
  ### 1) Make sure you can load your model and tokenizer using AutoClasses:
 
195
  ### 5) Select the correct precision
196
  Not all models are converted properly from `float16` to `bfloat16`, and selecting the wrong precision can sometimes cause evaluation error (as loading a `bf16` model in `fp16` can sometimes generate NaNs, depending on the weight range).
197
 
198
+ ## Model types
199
+ {icons}
 
 
200
  """
201
 
202
  CITATION_BUTTON_LABEL = "Copy the following snippet to cite these results"
src/display/utils.py CHANGED
@@ -120,6 +120,7 @@ class ModelDetails:
120
 
121
  class ModelType(Enum):
122
  PT = ModelDetails(name="pretrained", symbol="🟒")
 
123
  FT = ModelDetails(name="fine-tuned on domain-specific datasets", symbol="πŸ”Ά")
124
  chat = ModelDetails(name="chat models (RLHF, DPO, IFT, ...)", symbol="πŸ’¬")
125
  merges = ModelDetails(name="base merges and moerges", symbol="🀝")
@@ -132,6 +133,8 @@ class ModelType(Enum):
132
  def from_str(type):
133
  if "fine-tuned" in type or "πŸ”Ά" in type:
134
  return ModelType.FT
 
 
135
  if "pretrained" in type or "🟒" in type:
136
  return ModelType.PT
137
  if any([k in type for k in ["instruction-tuned", "RL-tuned", "chat", "🟦", "β­•", "πŸ’¬"]]):
 
120
 
121
  class ModelType(Enum):
122
  PT = ModelDetails(name="pretrained", symbol="🟒")
123
+ CPT = ModelDetails(name="continuously pretrained", symbol="🟩")
124
  FT = ModelDetails(name="fine-tuned on domain-specific datasets", symbol="πŸ”Ά")
125
  chat = ModelDetails(name="chat models (RLHF, DPO, IFT, ...)", symbol="πŸ’¬")
126
  merges = ModelDetails(name="base merges and moerges", symbol="🀝")
 
133
  def from_str(type):
134
  if "fine-tuned" in type or "πŸ”Ά" in type:
135
  return ModelType.FT
136
+ if "continously pretrained" in type or "🟩" in type:
137
+ return ModelType.CPT
138
  if "pretrained" in type or "🟒" in type:
139
  return ModelType.PT
140
  if any([k in type for k in ["instruction-tuned", "RL-tuned", "chat", "🟦", "β­•", "πŸ’¬"]]):
src/submission/check_validity.py CHANGED
@@ -143,15 +143,15 @@ def get_model_tags(model_card, model: str):
143
  if model_card is None:
144
  return tags
145
  if model_card.data.tags:
146
- is_merge_from_metadata = "merge" in model_card.data.tags
147
- is_moe_from_metadata = "moe" in model_card.data.tags
148
- merge_keywords = ["merged model", "merge model"]
149
  # If the model is a merge but not saying it in the metadata, we flag it
150
  is_merge_from_model_card = any(keyword in model_card.text.lower() for keyword in merge_keywords)
151
  if is_merge_from_model_card or is_merge_from_metadata:
152
  tags.append("merge")
153
- if not is_merge_from_metadata:
154
- tags.append("flagged:undisclosed_merge")
155
  moe_keywords = ["moe", "mixtral"]
156
  is_moe_from_model_card = any(keyword in model_card.text.lower() for keyword in moe_keywords)
157
  is_moe_from_name = "moe" in model.lower().replace("/", "-").replace("_", "-").split("-")
 
143
  if model_card is None:
144
  return tags
145
  if model_card.data.tags:
146
+ is_merge_from_metadata = ("merge" in model_card.data.tags or "moerge" in model_card.data.tags)
147
+ is_moe_from_metadata = ("moe" in model_card.data.tags or "moerge" in model_card.data.tags)
148
+ merge_keywords = ["merged model", "merge model", "moerge"]
149
  # If the model is a merge but not saying it in the metadata, we flag it
150
  is_merge_from_model_card = any(keyword in model_card.text.lower() for keyword in merge_keywords)
151
  if is_merge_from_model_card or is_merge_from_metadata:
152
  tags.append("merge")
153
+ #if not is_merge_from_metadata:
154
+ # tags.append("flagged:undisclosed_merge")
155
  moe_keywords = ["moe", "mixtral"]
156
  is_moe_from_model_card = any(keyword in model_card.text.lower() for keyword in moe_keywords)
157
  is_moe_from_name = "moe" in model.lower().replace("/", "-").replace("_", "-").split("-")