Nathan Habib commited on
Commit
2119dda
1 Parent(s): df0b79f

add fix for moe models

Browse files
Files changed (1) hide show
  1. src/submission/check_validity.py +2 -1
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 model == "Qwen/Qwen1.5-32B":
 
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: