antiquality commited on
Commit
46dc612
·
verified ·
1 Parent(s): cfa60a8

Fixing the exception logic

Browse files
Files changed (1) hide show
  1. src/submission/check_validity.py +1 -1
src/submission/check_validity.py CHANGED
@@ -61,7 +61,7 @@ def is_model_on_hub(model_name: str, revision: str, token: str = None, trust_rem
61
  )
62
 
63
  except Exception as e:
64
- return False, "was not found on hub!", None
65
 
66
 
67
  def get_model_size(model_info: ModelInfo, precision: str):
 
61
  )
62
 
63
  except Exception as e:
64
+ return False, f" error: {e}", None
65
 
66
 
67
  def get_model_size(model_info: ModelInfo, precision: str):