Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Nathan Habib
commited on
Commit
•
2119dda
1
Parent(s):
df0b79f
add fix for moe models
Browse files
src/submission/check_validity.py
CHANGED
@@ -151,7 +151,8 @@ def get_model_tags(model_card, model: str):
|
|
151 |
tags.append("merge")
|
152 |
is_moe_from_model_card = any(keyword in model_card.text.lower() for keyword in ["moe", "mixtral"])
|
153 |
# Hardcoding because of gating problem
|
154 |
-
if
|
|
|
155 |
is_moe_from_model_card = False
|
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:
|
|
|
151 |
tags.append("merge")
|
152 |
is_moe_from_model_card = any(keyword in model_card.text.lower() for keyword in ["moe", "mixtral"])
|
153 |
# Hardcoding because of gating problem
|
154 |
+
if "Qwen/Qwen1.5-32B" in model:
|
155 |
+
print("HERE NSHJNKJSNJLAS")
|
156 |
is_moe_from_model_card = False
|
157 |
is_moe_from_name = "moe" in model.lower().replace("/", "-").replace("_", "-").split("-")
|
158 |
if is_moe_from_model_card or is_moe_from_name or is_moe_from_metadata:
|