Clémentine commited on
Commit
a6f1b1f
1 Parent(s): 7dd994f

add mixtral to valid moe tags

Browse files
src/scripts/update_all_request_files.py CHANGED
@@ -59,7 +59,7 @@ def update_models(file_path, models):
59
  tags.append("merge")
60
  if not is_merge_from_metadata:
61
  tags.append("flagged:undisclosed_merge")
62
- moe_keywords = ["moe", "mixture of experts"]
63
  is_moe_from_model_card = any(keyword in model_card.text.lower() for keyword in moe_keywords)
64
  is_moe_from_name = "moe" in model_id.lower().replace("/", "-").replace("_", "-").split("-")
65
  if is_moe_from_model_card or is_moe_from_name or is_moe_from_metadata:
 
59
  tags.append("merge")
60
  if not is_merge_from_metadata:
61
  tags.append("flagged:undisclosed_merge")
62
+ moe_keywords = ["moe", "mixture of experts", "mixtral"]
63
  is_moe_from_model_card = any(keyword in model_card.text.lower() for keyword in moe_keywords)
64
  is_moe_from_name = "moe" in model_id.lower().replace("/", "-").replace("_", "-").split("-")
65
  if is_moe_from_model_card or is_moe_from_name or is_moe_from_metadata:
src/submission/submit.py CHANGED
@@ -116,7 +116,7 @@ def add_new_eval(
116
  tags.append("merge")
117
  if not is_merge_from_metadata:
118
  tags.append("flagged:undisclosed_merge")
119
- moe_keywords = ["moe", "mixture of experts"]
120
  is_moe_from_model_card = any(keyword in model_card.text.lower() for keyword in moe_keywords)
121
  is_moe_from_name = "moe" in model.lower().replace("/", "-").replace("_", "-").split("-")
122
  if is_moe_from_model_card or is_moe_from_name or is_moe_from_metadata:
 
116
  tags.append("merge")
117
  if not is_merge_from_metadata:
118
  tags.append("flagged:undisclosed_merge")
119
+ moe_keywords = ["moe", "mixture of experts", "mixtral"]
120
  is_moe_from_model_card = any(keyword in model_card.text.lower() for keyword in moe_keywords)
121
  is_moe_from_name = "moe" in model.lower().replace("/", "-").replace("_", "-").split("-")
122
  if is_moe_from_model_card or is_moe_from_name or is_moe_from_metadata: