Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
antiquality
commited on
Fixing the exception logic
Browse files
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, "
|
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):
|