Spaces:
Runtime error
Runtime error
spichlerz org
Browse files
src/leaderboard/read_evals.py
CHANGED
|
@@ -47,7 +47,12 @@ class EvalResult:
|
|
| 47 |
|
| 48 |
# Get model and org
|
| 49 |
org_and_model = config.get("model_name", config.get("model_args", None))
|
| 50 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
org_and_model = org_and_model.split("/", 1)
|
| 52 |
|
| 53 |
if len(org_and_model) == 1:
|
|
|
|
| 47 |
|
| 48 |
# Get model and org
|
| 49 |
org_and_model = config.get("model_name", config.get("model_args", None))
|
| 50 |
+
SPICHLERZ_ORG = "spichlerz/"
|
| 51 |
+
|
| 52 |
+
if re.match(r"^pretrained=(.*/(plgkwrobel|plggspkl)/)(models/)?", org_and_model):
|
| 53 |
+
org_and_model = re.sub(r"^pretrained=(.*/(plgkwrobel|plggspkl)/)(models/)?", SPICHLERZ_ORG, org_and_model)
|
| 54 |
+
|
| 55 |
+
org_and_model = re.sub(r"^pretrained=", "", org_and_model)
|
| 56 |
org_and_model = org_and_model.split("/", 1)
|
| 57 |
|
| 58 |
if len(org_and_model) == 1:
|