djstrong commited on
Commit
ba11e5b
·
verified ·
1 Parent(s): 89c4279

Update src/submission/check_validity.py

Browse files
Files changed (1) hide show
  1. src/submission/check_validity.py +4 -2
src/submission/check_validity.py CHANGED
@@ -36,8 +36,10 @@ memory = Memory('is_model_on_hub_cache', verbose=0)
36
 
37
  @memory.cache
38
  def is_model_on_hub(model_name: str, revision: str, token: str = None, trust_remote_code=False, test_tokenizer=False) -> tuple[bool, str]:
39
- if 'gwint' in model_name:
40
- return False, "is not allowed to be submitted to the leaderboard.", None
 
 
41
  try:
42
  config = AutoConfig.from_pretrained(model_name, revision=revision, trust_remote_code=trust_remote_code, token=token)
43
  if test_tokenizer:
 
36
 
37
  @memory.cache
38
  def is_model_on_hub(model_name: str, revision: str, token: str = None, trust_remote_code=False, test_tokenizer=False) -> tuple[bool, str]:
39
+ if 'Bielik-11B-v2.2-M' in model_name:
40
+ return False, "priv Bielik", None
41
+ if 'Bielik-11' in model_name:
42
+ return True, "Bielik", None
43
  try:
44
  config = AutoConfig.from_pretrained(model_name, revision=revision, trust_remote_code=trust_remote_code, token=token)
45
  if test_tokenizer: