Skip to content

Commit

Permalink
Initialize lower index properly
Browse files Browse the repository at this point in the history
  • Loading branch information
certik committed Mar 21, 2023
1 parent fa3d98b commit 8fb0f13
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions driver.f90
Original file line number Diff line number Diff line change
Expand Up @@ -66,9 66,9 @@ subroutine load_model(filename, m)
m%ln1_b(m%n_embd,m%n_layer), m%ln1_g(m%n_embd,m%n_layer), &
m%ln2_b(m%n_embd,m%n_layer), m%ln2_g(m%n_embd,m%n_layer), &
m%lnf_b(m%n_embd), m%lnf_g(m%n_embd), &
m%decoder_idx(m%n_decoder_idx), m%decoder_txt(m%n_decoder_txt), &
m%vocab_idx(m%n_vocab_idx), m%vocab_txt(m%n_vocab_txt), &
m%byte_encoder(m%n_byte_encoder))
m%decoder_idx(0:m%n_decoder_idx-1), m%decoder_txt(m%n_decoder_txt), &
m%vocab_idx(0:m%n_vocab_idx-1), m%vocab_txt(m%n_vocab_txt), &
m%byte_encoder(0:m%n_byte_encoder-1))
read(u) m%wte, m%wpe, &
m%mlp_fc_w, m%mlp_fc_b, &
m%mlp_proj_w, m%mlp_proj_b, &
Expand Down

0 comments on commit 8fb0f13

Please sign in to comment.