jpmschweitzer/odysseus
1
0
Fork 0
mirror of https://github.com/pewdiepie-archdaemon/odysseus.git synced 2026-10-31 10:12:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #6730 from dxrpl/fix/hwfit-hf-param-count

Browse Source 
fix(hwfit): keep the MoE active size for HF collection models (#6729)
This commit is contained in:
Alexandre Teixeira
2026-10-30 19:43:34 +00:00
committed by GitHub
parent e7796f71d3 bfa514f6e8
commit b88154ecf9
2 changed files with 37 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
services/hwfit/hf_discovery.py
+4 -2
View File
@@ -226,9 +226,11 @@ def _entry_from_collection_item(collection, item, source):
if repo_prefix and not repo_id.startswith(repo_prefix):
return None
raw_params = item.get("numParameters") or 0
active = None
# Read the name even when HF gives a total: it is the only source of the
# MoE active size (e.g. "-35B-A3B"), which the speed estimate depends on.
name_params, active = _parse_params_from_name(repo_id)
if not raw_params:
raw_params, active = _parse_params_from_name(repo_id)
raw_params = name_params
param_label, raw_params = _format_params(raw_params)
if not raw_params:
return None