Spaces:
Sleeping
Sleeping
Quentin Gallouédec
commited on
Commit
·
948a5b8
1
Parent(s):
c44db38
support gated models
Browse files- src/backend.py +8 -3
src/backend.py
CHANGED
|
@@ -537,10 +537,15 @@ def _backend_routine():
|
|
| 537 |
row = {"model_id": model_id, "user_id": user_id, "sha": sha}
|
| 538 |
|
| 539 |
# Run an evaluation on the models
|
| 540 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 541 |
|
| 542 |
-
# Extract the environment IDs from the tags (usually only one)
|
| 543 |
-
env_ids = pattern_match(model_info.tags, ALL_ENV_IDS)
|
| 544 |
if len(env_ids) > 0:
|
| 545 |
env = env_ids[0]
|
| 546 |
logger.info(f"Running evaluation on {user_id}/{model_id}")
|
|
|
|
| 537 |
row = {"model_id": model_id, "user_id": user_id, "sha": sha}
|
| 538 |
|
| 539 |
# Run an evaluation on the models
|
| 540 |
+
try:
|
| 541 |
+
model_info = API.model_info(repo_id, revision=sha)
|
| 542 |
+
# Extract the environment IDs from the tags (usually only one)
|
| 543 |
+
env_ids = pattern_match(model_info.tags, ALL_ENV_IDS)
|
| 544 |
+
except Exception as e:
|
| 545 |
+
logger.error(f"Error fetching model info for {repo_id}: {e}")
|
| 546 |
+
logger.exception(e)
|
| 547 |
+
env_ids = []
|
| 548 |
|
|
|
|
|
|
|
| 549 |
if len(env_ids) > 0:
|
| 550 |
env = env_ids[0]
|
| 551 |
logger.info(f"Running evaluation on {user_id}/{model_id}")
|