Skip to content

Commit 634e5a9

Browse files
DamonFoolCISC
andauthored
Update src/llama-model.cpp
Co-authored-by: Sigbjørn Skjæret <[email protected]>
1 parent 12a909f commit 634e5a9

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/llama-model.cpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4420,7 +4420,9 @@ bool llama_model::load_tensors(llama_model_loader & ml) {
44204420
// n_layer: number of encoder_layers
44214421
// n_dec_layer: number of decoder_layers
44224422
const int n_dec_layer = hparams.n_dec_layer;
4423-
layers.resize(n_layer + n_dec_layer);
4423+
if (n_dec_layer > n_layer) {
4424+
layers.resize(n_dec_layer);
4425+
}
44244426

44254427
// load encoder layers
44264428
for (int i = 0; i < n_layer; ++i) {

0 commit comments

Comments
 (0)