djstrong commited on
Commit
dc92b05
1 Parent(s): e066704

update cache

Browse files
Files changed (1) hide show
  1. src/submission/check_validity.py +2 -0
src/submission/check_validity.py CHANGED
@@ -36,6 +36,8 @@ 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 'Bielik-11' in model_name:
40
  return True, "Bielik", None
41
  try:
 
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: