Skip to content

Commit 881cca0

Browse files
committed
fix: loader.
1 parent c2792a1 commit 881cca0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/llm.cpp

+2-2
Original file line numberDiff line numberDiff line change
@@ -467,9 +467,9 @@ void loadLlmNetWeight(const char *path, LlmNet *net, NnRootWeightLoader *loader)
467467
b += loader->loadAll("final_rms_norm", 0, net->rmsNormSize.nBytes, b);
468468
b += loader->loadRowMatmulSlices("final_matmul_logits", 0, &net->wclsSlice, b);
469469

470-
loader->finish();
471-
472470
unsigned long missingBytes = (b - data) - net->header->fileSize;
473471
assert(missingBytes == 0);
474472
printf("💿 Weights loaded\n");
473+
474+
loader->finish();
475475
}

0 commit comments

Comments
 (0)