Atomic-Germ commited on
Commit
fd5a58e
·
verified ·
1 Parent(s): dcfcf69

flm-add.py: fix -A3B size suffix (35b-a3b) and add family aliases (granite/llama, darwin/qwen3.6-moe, crow/qwythos/qwopus/huihui/qwen3.5, gemma-4/gemma4e)

Browse files
Files changed (1) hide show
  1. flm-add.py +11 -1
flm-add.py CHANGED
@@ -64,8 +64,16 @@ FAMILY_ALIASES = [
64
  ("qwen2vl", "qwen2vl"),
65
  ("qwen2", "qwen2"),
66
  ("gemma4", "gemma4e"),
 
67
  ("gemma3", "gemma3"),
68
  ("llama3", "llama3"),
 
 
 
 
 
 
 
69
  ("deepseek-r1-0528", "deepseek-r1-0528"),
70
  ("deepseek-r1", "deepseek-r1"),
71
  ("deepseek", "deepseek-r1"),
@@ -164,7 +172,9 @@ def _strip_npu2(name):
164
 
165
 
166
  def _extract_size(bare):
167
- m = re.search(r"(\d+(?:\.\d+)?[Bb](?:-[A-Za-z]+\d+(?:\.\d+)?)*)", bare)
 
 
168
  if not m:
169
  return None, bare
170
  size = m.group(1).lower()
 
64
  ("qwen2vl", "qwen2vl"),
65
  ("qwen2", "qwen2"),
66
  ("gemma4", "gemma4e"),
67
+ ("gemma-4", "gemma4e"),
68
  ("gemma3", "gemma3"),
69
  ("llama3", "llama3"),
70
+ ("llama", "llama3"),
71
+ ("granite", "llama3"),
72
+ ("crow", "qwen3.5"),
73
+ ("huihui", "qwen3.5"),
74
+ ("qwythos", "qwen3.5"),
75
+ ("qwopus", "qwen3.5"),
76
+ ("darwin", "qwen3.6-moe"),
77
  ("deepseek-r1-0528", "deepseek-r1-0528"),
78
  ("deepseek-r1", "deepseek-r1"),
79
  ("deepseek", "deepseek-r1"),
 
172
 
173
 
174
  def _extract_size(bare):
175
+ # Trailing size groups like "-A3B" (Qwen3.6-35B-A3B) end in a letter that
176
+ # the digit group must not swallow (previously "-A3B" -> "35b-a3").
177
+ m = re.search(r"(\d+(?:\.\d+)?[Bb](?:-[A-Za-z]+\d+(?:\.\d+)?[A-Za-z]*)*)", bare)
178
  if not m:
179
  return None, bare
180
  size = m.group(1).lower()