Skip to content

Commit

Permalink
Fix default value when multiquery option is missing
Browse files Browse the repository at this point in the history
  • Loading branch information
guillaumekln committed Jul 12, 2023
1 parent 119f1c9 commit 4175e4d
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions python/ctranslate2/converters/opennmt_py.py
Original file line number Diff line number Diff line change
Expand Up @@ -82,7 +82,7 @@ def _get_model_spec_seq2seq(
alignment_heads=alignment_heads,
num_source_embeddings=num_source_embeddings,
embeddings_merge=_SUPPORTED_FEATURES_MERGE[feat_merge],
multi_query_attention=getattr(opt, "multiquery", True),
multi_query_attention=getattr(opt, "multiquery", False),
)

model_spec.config.decoder_start_token = getattr(opt, "decoder_start_token", "<s>")
Expand Down Expand Up @@ -116,7 +116,7 @@ def _get_model_spec_lm(opt, variables, src_vocabs, tgt_vocabs, num_source_embedd
rms_norm=opt.layer_norm == "rms",
rotary_dim=rotary_dim,
rotary_interleave=True,
multi_query_attention=getattr(opt, "multiquery", True),
multi_query_attention=getattr(opt, "multiquery", False),
)

set_transformer_decoder(
Expand Down

0 comments on commit 4175e4d

Please sign in to comment.