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- 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 |
-
|
|
|
|
|
|
|
| 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()
|