aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--main.cpp10
1 files changed, 1 insertions, 9 deletions
diff --git a/main.cpp b/main.cpp
index eb78fe5..c88405b 100644
--- a/main.cpp
+++ b/main.cpp
@@ -143,16 +143,8 @@ bool llama_model_load(const std::string & fname, llama_model & model, gpt_vocab
// load vocab
{
- const int32_t n_vocab = model.hparams.n_vocab;
-
- if (n_vocab != model.hparams.n_vocab) {
- fprintf(stderr, "%s: invalid model file '%s' (bad vocab size %d != %d)\n",
- __func__, fname.c_str(), n_vocab, model.hparams.n_vocab);
- return false;
- }
-
std::string word;
- for (int i = 0; i < n_vocab; i++) {
+ for (int i = 0; i < model.hparams.n_vocab; i++) {
uint32_t len;
fin.read((char *) &len, sizeof(len));