Clémentine commited on
Commit
bcf0226
1 Parent(s): f04f90e

relaxed filters on merged models

Browse files
Files changed (1) hide show
  1. src/submission/check_validity.py +2 -2
src/submission/check_validity.py CHANGED
@@ -144,14 +144,14 @@ def get_model_tags(model_card, model: str):
144
  if model_card.data.tags:
145
  is_merge_from_metadata = "merge" in model_card.data.tags
146
  is_moe_from_metadata = "moe" in model_card.data.tags
147
- merge_keywords = ["mergekit", "merged model", "merge model", "merging"]
148
  # If the model is a merge but not saying it in the metadata, we flag it
149
  is_merge_from_model_card = any(keyword in model_card.text.lower() for keyword in merge_keywords)
150
  if is_merge_from_model_card or is_merge_from_metadata:
151
  tags.append("merge")
152
  if not is_merge_from_metadata:
153
  tags.append("flagged:undisclosed_merge")
154
- moe_keywords = ["moe", "mixture of experts", "mixtral"]
155
  is_moe_from_model_card = any(keyword in model_card.text.lower() for keyword in moe_keywords)
156
  is_moe_from_name = "moe" in model.lower().replace("/", "-").replace("_", "-").split("-")
157
  if is_moe_from_model_card or is_moe_from_name or is_moe_from_metadata:
 
144
  if model_card.data.tags:
145
  is_merge_from_metadata = "merge" in model_card.data.tags
146
  is_moe_from_metadata = "moe" in model_card.data.tags
147
+ merge_keywords = ["merged model", "merge model"]
148
  # If the model is a merge but not saying it in the metadata, we flag it
149
  is_merge_from_model_card = any(keyword in model_card.text.lower() for keyword in merge_keywords)
150
  if is_merge_from_model_card or is_merge_from_metadata:
151
  tags.append("merge")
152
  if not is_merge_from_metadata:
153
  tags.append("flagged:undisclosed_merge")
154
+ moe_keywords = ["moe", "mixtral"]
155
  is_moe_from_model_card = any(keyword in model_card.text.lower() for keyword in moe_keywords)
156
  is_moe_from_name = "moe" in model.lower().replace("/", "-").replace("_", "-").split("-")
157
  if is_moe_from_model_card or is_moe_from_name or is_moe_from_metadata: