include llama.cpp since there were some edits
This commit is contained in:
1 parent
db7f28d087
commit
8c66990853
3918 files changed
+1640170
-1
No files matched your search
@@ -0,0 +1,54 @@
|
||||
import { groupModelFamilies, modelFamilyKey } from '$lib/utils/model-families';
|
||||
import { describe, expect, it } from 'vitest';
|
||||
|
||||
describe('modelFamilyKey', () => {
|
||||
it('folds the leading letters of the model name', () => {
|
||||
expect(modelFamilyKey('Qwen3.5-27B')).toBe('Qwen');
|
||||
expect(modelFamilyKey('Qwen3.8-27B-GGUF')).toBe('Qwen');
|
||||
expect(modelFamilyKey('Llama-3.2-3B-Instruct')).toBe('Llama');
|
||||
});
|
||||
|
||||
it('uses the model and not the org', () => {
|
||||
expect(modelFamilyKey('ggml-org/Qwen3.8-27B-GGUF')).toBe('Qwen');
|
||||
});
|
||||
|
||||
it('falls back to the first segment when the name starts with a digit', () => {
|
||||
expect(modelFamilyKey('3.5-MoE-instruct')).toBe('3');
|
||||
});
|
||||
|
||||
it('keeps a name with no letters whole', () => {
|
||||
expect(modelFamilyKey('_internal')).toBe('_internal');
|
||||
expect(modelFamilyKey('-leading-dash')).toBe('-leading-dash');
|
||||
});
|
||||
});
|
||||
|
||||
describe('groupModelFamilies', () => {
|
||||
const rows = [
|
||||
{ name: 'ggml-org/Qwen3.5-9B-GGUF' },
|
||||
{ name: 'unsloth/Qwen3.8-27B-GGUF' },
|
||||
{ name: 'meta-llama/Llama-3.2-3B-Instruct' },
|
||||
{ name: 'org/tinyllama-1.1B' }
|
||||
];
|
||||
const modelOf = (row: { name: string }) => row.name;
|
||||
|
||||
it('collects the entries of one family', () => {
|
||||
const families = groupModelFamilies(rows, modelOf);
|
||||
|
||||
expect(families.map((family) => [family.key, family.entries.length])).toEqual([
|
||||
['Qwen', 2],
|
||||
['Llama', 1],
|
||||
['tinyllama', 1]
|
||||
]);
|
||||
});
|
||||
|
||||
it('labels a family with its key', () => {
|
||||
expect(groupModelFamilies(rows, modelOf)[0].label).toBe('Qwen');
|
||||
});
|
||||
|
||||
it('keeps the input order of the entries', () => {
|
||||
expect(groupModelFamilies(rows, modelOf)[0].entries.map(modelOf)).toEqual([
|
||||
'ggml-org/Qwen3.5-9B-GGUF',
|
||||
'unsloth/Qwen3.8-27B-GGUF'
|
||||
]);
|
||||
});
|
||||
});
|
||||
Reference in new issue
Block a user