Skip to content

Commit eefe245

Browse files
committed
Fix quantized sizes not showing in collapsed family headers
When "Show quantized models" is enabled, collapsed family headers were only showing base sizes (e.g., "27B") instead of including quantized variants (e.g., "27B-Q4"). The bug was in CatalogSection.buildCatalogItems() where size collection used model.size instead of model.sizeLabel. Since model.size returns just the base size for all models, quantized variants weren't distinguished in the header. Now uses model.sizeLabel which includes the quantization suffix for quantized models.
1 parent fa3f053 commit eefe245

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

LlamaBarn/Menu/Sections.swift

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -281,8 +281,8 @@ final class CatalogSection {
281281
// Group models by family to collect unique sizes
282282
var familySizes: [String: [String]] = [:]
283283
for model in sortedModels {
284-
if !familySizes[model.family, default: []].contains(model.size) {
285-
familySizes[model.family, default: []].append(model.size)
284+
if !familySizes[model.family, default: []].contains(model.sizeLabel) {
285+
familySizes[model.family, default: []].append(model.sizeLabel)
286286
}
287287
}
288288

0 commit comments

Comments
 (0)