Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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:
|