Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
shigeki Ishida
commited on
Commit
·
b463be8
1
Parent(s):
67542c9
Restore missing implementation of the architecture variable.
Browse files- src/submission/submit.py +9 -1
src/submission/submit.py
CHANGED
@@ -48,7 +48,14 @@ def add_new_eval(
|
|
48 |
revision = "main"
|
49 |
|
50 |
# Is the model on the hub?
|
51 |
-
model_on_hub, error,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
52 |
if not model_on_hub:
|
53 |
return styled_error(f'Model "{model}" {error}')
|
54 |
|
@@ -86,6 +93,7 @@ def add_new_eval(
|
|
86 |
"private": False,
|
87 |
"add_special_tokens": add_special_tokens,
|
88 |
"llm_jp_eval_version": current_version,
|
|
|
89 |
"vllm_version": current_vllm_version,
|
90 |
}
|
91 |
|
|
|
48 |
revision = "main"
|
49 |
|
50 |
# Is the model on the hub?
|
51 |
+
model_on_hub, error, model_config = is_model_on_hub(
|
52 |
+
model_name=model, revision=revision, token=TOKEN, test_tokenizer=True
|
53 |
+
)
|
54 |
+
architecture = "?"
|
55 |
+
if model_config is not None:
|
56 |
+
architectures = getattr(model_config, "architectures", None)
|
57 |
+
if architectures:
|
58 |
+
architecture = ";".join(architectures)
|
59 |
if not model_on_hub:
|
60 |
return styled_error(f'Model "{model}" {error}')
|
61 |
|
|
|
93 |
"private": False,
|
94 |
"add_special_tokens": add_special_tokens,
|
95 |
"llm_jp_eval_version": current_version,
|
96 |
+
"architecture": architecture,
|
97 |
"vllm_version": current_vllm_version,
|
98 |
}
|
99 |
|