diff --git a/convert.py b/convert.py index 5816a72..b3f68cb 100644 --- a/convert.py +++ b/convert.py @@ -155,9 +155,9 @@ def dump_config(f, config, ggml_type): config.num_layers, config.inner_hidden_size, config.max_sequence_length, - config.bos_token_id, - config.eos_token_id, - config.pad_token_id, + config.bos_token_id if config.bos_token_id is not None else -1, + config.eos_token_id if config.eos_token_id is not None else -1, + config.pad_token_id if config.pad_token_id is not None else -1, config.sep_token_id if config.sep_token_id is not None else -1, ] f.write(struct.pack("i" * len(config_values), *config_values)) @@ -224,8 +224,8 @@ def dump_config(f, config, ggml_type): config.ffn_hidden_size, config.seq_length, config.bos_token_id if config.bos_token_id is not None else -1, - config.eos_token_id, - config.pad_token_id, + config.eos_token_id if config.eos_token_id is not None else -1, + config.pad_token_id if config.pad_token_id is not None else -1, config.sep_token_id if config.sep_token_id is not None else -1, config.multi_query_group_num, ]