Skip to content

Commit 978b995

Browse files
committed
[gguf] ad group prefix to RE_GGUF_SHARD_FILE
1 parent 24fa435 commit 978b995

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

packages/gguf/src/gguf.spec.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -226,6 +226,7 @@ describe("gguf", () => {
226226
const match = ggufPath.match(RE_GGUF_SHARD_FILE);
227227

228228
expect(RE_GGUF_SHARD_FILE.test(ggufPath)).toEqual(true);
229+
expect(match?.groups?.prefix).toEqual("grok-1/grok-1-q4_0");
229230
expect(match?.groups?.shard).toEqual("00003");
230231
expect(match?.groups?.total).toEqual("00009");
231232
});

packages/gguf/src/gguf.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ export { GGUFValueType, GGMLQuantizationType } from "./types";
66
export { GGUF_QUANT_DESCRIPTIONS } from "./quant-descriptions";
77

88
export const RE_GGUF_FILE = /\.gguf$/;
9-
export const RE_GGUF_SHARD_FILE = /-(?<shard>\d{5})-of-(?<total>\d{5})\.gguf$/;
9+
export const RE_GGUF_SHARD_FILE = /^(?<prefix>.*?)-(?<shard>\d{5})-of-(?<total>\d{5})\.gguf$/;
1010

1111
const isVersion = (version: number): version is Version => version === 1 || version === 2 || version === 3;
1212

0 commit comments

Comments
 (0)