Skip to content

Commit 0493606

Browse files
committed
load tokenizer before model.
1 parent 9c3b964 commit 0493606

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/main.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -297,10 +297,10 @@ int run(ProgramArgs* args, void (*program)(Inference* inference, SocketPool* soc
297297
args->steps = spec.seqLen;
298298
}
299299

300+
Tokenizer tokenizer(args->tokenizerPath, spec.vocabSize);
300301
Transformer transformer = Transformer::loadRootFromFile(args->modelPath, &spec, socketPool);
301302
Inference inference = Inference(&arch, args->nThreads, &transformer, socketPool);
302303

303-
Tokenizer tokenizer(args->tokenizerPath, spec.vocabSize);
304304
Sampler sampler(spec.vocabSize, args->temperature, args->topp, rngSeed);
305305

306306
program(&inference, socketPool, &tokenizer, &sampler, args, &spec);

0 commit comments

Comments
 (0)