Skip to content

Commit e1177fc

Browse files
committed
@LysandreJik added an actual model
1 parent c0fc9f3 commit e1177fc

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

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

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -122,16 +122,15 @@ describe("parseSafetensorsMetadata", () => {
122122
// total params = 109M
123123
});
124124

125-
it.skip("fetch info for single-file, but get param count directly from metadata", async () => {
126-
/// we don't have an example for this on the Hub yet... cc @LysandreJik
125+
it("fetch info for single-file, but get param count directly from metadata", async () => {
127126
const parse = await parseSafetensorsMetadata({
128-
repo: "hf-internal-testing/non-sharded-model",
127+
repo: "hf-internal-testing/single-file-model",
129128
computeParametersCount: true,
130-
revision: "ce6373360e61e6f70b4a1e0cfcc9407b008dea5b",
129+
revision: "75fcd3fed0285ac7f1092897ff2aefdf24bf872e",
131130
});
132131

133132
assert(!parse.sharded);
134-
assert.deepStrictEqual(parse.parameterTotal, 666);
133+
assert.deepStrictEqual(parse.parameterTotal, 109_482_240);
135134
});
136135

137136
it("should detect sharded safetensors filename", async () => {

0 commit comments

Comments
 (0)