Clémentine
commited on
Commit
•
bcf0226
1
Parent(s):
f04f90e
relaxed filters on merged models
Browse files
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 = ["
|
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", "
|
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:
|