From 7afc394b7d4712eb1f10311f2008b916c8a81228 Mon Sep 17 00:00:00 2001 From: minhthuc Date: Fri, 20 Sep 2024 11:32:49 +0200 Subject: [PATCH] fix black --- python/ctranslate2/converters/transformers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/ctranslate2/converters/transformers.py b/python/ctranslate2/converters/transformers.py index 317b2d50c..f63d56c4c 100644 --- a/python/ctranslate2/converters/transformers.py +++ b/python/ctranslate2/converters/transformers.py @@ -1876,7 +1876,7 @@ def get_model_spec(self, model): quant_type=quant_type, quant_group_size=quant_group_size, quant_bits=quant_bits, - head_dim=model.config.head_dim + head_dim=model.config.head_dim, ) self.set_decoder(spec.decoder, model.model, quant_type=quant_type)