Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Minseok Bae
commited on
Commit
•
3b66490
1
Parent(s):
0c85a8e
modified is_model_on_hub()
Browse files
src/submission/check_validity.py
CHANGED
@@ -54,7 +54,8 @@ def is_model_on_hub(model_name: str, revision: str, token: str = None, trust_rem
|
|
54 |
)
|
55 |
|
56 |
except Exception as e:
|
57 |
-
return False, "was not found on hub
|
|
|
58 |
|
59 |
|
60 |
def get_model_size(model_info: ModelInfo, precision: str):
|
|
|
54 |
)
|
55 |
|
56 |
except Exception as e:
|
57 |
+
return False, f"was not found on hub!: {e}", None
|
58 |
+
|
59 |
|
60 |
|
61 |
def get_model_size(model_info: ModelInfo, precision: str):
|