CompactAI commited on
Commit
5eb4a0e
·
verified ·
1 Parent(s): 5ca1b9e

Upload 2 files

Browse files
Files changed (1) hide show
  1. app.py +13 -2
app.py CHANGED
@@ -2910,7 +2910,9 @@ def _compare_fn(prompt, selected_versions, mode_key, use_custom,
2910
  cfg["max_new_tokens"] = int(max_tokens)
2911
  cfg["context_window"] = int(ctx_win)
2912
 
2913
- all_versions = _collection_versions()
 
 
2914
  selected = set(selected_versions or [])
2915
  state = {v: ("⏳ Queued…" if v in selected else "") for v in all_versions}
2916
 
@@ -3060,7 +3062,16 @@ def _advanced_block():
3060
 
3061
  # ---- build UI ----
3062
 
3063
- _initial_versions = _collection_versions()
 
 
 
 
 
 
 
 
 
3064
  _initial_version = _initial_versions[0] if _initial_versions else None
3065
  _initial_ckpt_labels = _ckpt_labels(_initial_version) if _initial_version else []
3066
  _mode_keys = list(MODES.keys())
 
2910
  cfg["max_new_tokens"] = int(max_tokens)
2911
  cfg["context_window"] = int(ctx_win)
2912
 
2913
+ # Iterate + emit oldest → newest (Haiku-1 first, Glint-1 last) so the order
2914
+ # matches the output-box layout in the UI.
2915
+ all_versions = _sort_oldest_to_newest(_collection_versions())
2916
  selected = set(selected_versions or [])
2917
  state = {v: ("⏳ Queued…" if v in selected else "") for v in all_versions}
2918
 
 
3062
 
3063
  # ---- build UI ----
3064
 
3065
+ def _sort_oldest_to_newest(versions: list[str]) -> list[str]:
3066
+ """Sort versions oldest→newest using HUGGINGFACE_MODELS key order."""
3067
+ order = {name: i for i, name in enumerate(HUGGINGFACE_MODELS)}
3068
+ return sorted(
3069
+ versions,
3070
+ key=lambda v: (order.get(v, len(order)), versions.index(v)),
3071
+ )
3072
+
3073
+
3074
+ _initial_versions = _sort_oldest_to_newest(_collection_versions())
3075
  _initial_version = _initial_versions[0] if _initial_versions else None
3076
  _initial_ckpt_labels = _ckpt_labels(_initial_version) if _initial_version else []
3077
  _mode_keys = list(MODES.keys())