From 72da51f1fe4bc26ea13dbc7efaa284361771c4b1 Mon Sep 17 00:00:00 2001 From: Minh-Thuc <46375464+minhthuc2502@users.noreply.github.com> Date: Tue, 28 Nov 2023 12:46:47 +0100 Subject: [PATCH] check alignment_heads in gen config before using (#1565) Co-authored-by: thucpham --- python/ctranslate2/converters/transformers.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/ctranslate2/converters/transformers.py b/python/ctranslate2/converters/transformers.py index c5a674967..1b2d0c98f 100644 --- a/python/ctranslate2/converters/transformers.py +++ b/python/ctranslate2/converters/transformers.py @@ -915,7 +915,8 @@ def set_config(self, config, model, tokenizer): if gen_config is not None: config.suppress_ids = gen_config.suppress_tokens config.suppress_ids_begin = gen_config.begin_suppress_tokens - config.alignment_heads = gen_config.alignment_heads + if hasattr(gen_config, "alignment_heads"): + config.alignment_heads = gen_config.alignment_heads if hasattr(gen_config, "lang_to_id"): config.lang_ids = sorted(gen_config.lang_to_id.values()) else: