Skip to content

Commit 09d8032

Browse files
author
Mishig
authored
[safetensors parser] RE_SAFETENSORS_SHARD_FILE (#593)
```ts const RE_SAFETENSORS_SHARD_FILE = /\d{5}-of-\d{5}\.safetensors$/; ``` is this regex good to detect shard file based on safetensors filename ?
1 parent 32e3e48 commit 09d8032

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

packages/hub/src/lib/parse-safetensors-metadata.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ export const SAFETENSORS_INDEX_FILE = "model.safetensors.index.json";
1414
/// but in some situations safetensors weights have different filenames.
1515
export const RE_SAFETENSORS_FILE = /\.safetensors$/;
1616
export const RE_SAFETENSORS_INDEX_FILE = /\.safetensors\.index\.json$/;
17+
export const RE_SAFETENSORS_SHARD_FILE = /\d{5}-of-\d{5}\.safetensors$/;
1718
const PARALLEL_DOWNLOADS = 5;
1819
const MAX_HEADER_LENGTH = 25_000_000;
1920

0 commit comments

Comments
 (0)