Jacqueline Garrahan commited on
Commit
6f9abbe
·
unverified ·
1 Parent(s): a548d11

Fix validity check

Browse files
Files changed (1) hide show
  1. src/submission/check_validity.py +1 -8
src/submission/check_validity.py CHANGED
@@ -35,9 +35,8 @@ def check_model_card(repo_id: str) -> tuple[bool, str]:
35
 
36
  def is_model_on_hub(model_name: str, token: str = None, trust_remote_code=False, test_tokenizer=False) -> tuple[bool, str]:
37
  """Checks if the model model_name is on the hub, and whether it (and its tokenizer) can be loaded with AutoClasses."""
38
- model_info = API.model_info(model_name, revision="main")
39
  try:
40
- model_info = API.model_info(model_name)
41
  config = model_info.config
42
  if test_tokenizer:
43
  try:
@@ -52,12 +51,6 @@ def is_model_on_hub(model_name: str, token: str = None, trust_remote_code=False,
52
  return (False, "'s tokenizer cannot be loaded. Is your tokenizer class in a stable transformers release, and correctly configured?", None)
53
  return True, None, config
54
 
55
- except ValueError:
56
- return (
57
- False,
58
- "needs to be launched with `trust_remote_code=True`. For safety reason, we do not allow these models to be automatically submitted to the leaderboard.",
59
- None
60
- )
61
 
62
  except Exception as e:
63
  return False, "was not found on hub!", None
 
35
 
36
  def is_model_on_hub(model_name: str, token: str = None, trust_remote_code=False, test_tokenizer=False) -> tuple[bool, str]:
37
  """Checks if the model model_name is on the hub, and whether it (and its tokenizer) can be loaded with AutoClasses."""
 
38
  try:
39
+ model_info = API.model_info(model_name, revision="main")
40
  config = model_info.config
41
  if test_tokenizer:
42
  try:
 
51
  return (False, "'s tokenizer cannot be loaded. Is your tokenizer class in a stable transformers release, and correctly configured?", None)
52
  return True, None, config
53
 
 
 
 
 
 
 
54
 
55
  except Exception as e:
56
  return False, "was not found on hub!", None