Fixing tests
This commit is contained in:
parent
81fae1dc8f
commit
281a4b4f27
2 changed files with 5 additions and 4 deletions
|
@ -995,6 +995,11 @@ static void llama_model_load_internal(
|
||||||
std::unique_ptr<llama_model_loader> ml(new llama_model_loader(fname, use_mmap));
|
std::unique_ptr<llama_model_loader> ml(new llama_model_loader(fname, use_mmap));
|
||||||
|
|
||||||
vocab = std::move(ml->file_loader->vocab);
|
vocab = std::move(ml->file_loader->vocab);
|
||||||
|
|
||||||
|
if (vocab_only) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
model.hparams = ml->file_loader->hparams;
|
model.hparams = ml->file_loader->hparams;
|
||||||
model.n_gpu_layers = n_gpu_layers;
|
model.n_gpu_layers = n_gpu_layers;
|
||||||
llama_file_version file_version = ml->file_loader->file_version;
|
llama_file_version file_version = ml->file_loader->file_version;
|
||||||
|
@ -1055,10 +1060,6 @@ static void llama_model_load_internal(
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (vocab_only) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
auto & ctx = model.ctx;
|
auto & ctx = model.ctx;
|
||||||
|
|
||||||
size_t ctx_size;
|
size_t ctx_size;
|
||||||
|
|
Binary file not shown.
Loading…
Add table
Add a link
Reference in a new issue