ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Make replacement logic conscise

Browse source
This commit is contained in:
datta0 2025-06-01 05:57:43 +00:00
commit 46b722842b
1 changed files with 5 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

11
unsloth/models/_utils.py
View file

@ -217,12 +217,11 @@ def extract_approx_params_from_config(config):
billions = float(match_B.group(1))
return int(1_000_000_000 * billions)
else:
for fam in lowercase_b_families:
if fam in cleaned.lower():
match_b = re.search(r"([0-9]+(?:\.[0-9]+)?)\s*b", cleaned)
if match_b:
billions = float(match_b.group(1))
return int(1_000_000_000 * billions)
if any(fam in cleaned.lower() for fam in lowercase_b_families):
match_b = re.search(r"([0-9]+(?:\.[0-9]+)?)\s*b", cleaned)
if match_b:
billions = float(match_b.group(1))
return int(1_000_000_000 * billions)
else:
match_any = re.search(r"([0-9]+(?:\.[0-9]+)?)\s*[bB]", cleaned)
if match_any: