shigeki Ishida commited on
Commit
b463be8
·
1 Parent(s): 67542c9

Restore missing implementation of the architecture variable.

Browse files
Files changed (1) hide show
  1. 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, _ = is_model_on_hub(model_name=model, revision=revision, token=TOKEN, test_tokenizer=True)
 
 
 
 
 
 
 
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