Update goai_helpers/goai_traduction.py

goai_helpers/goai_traduction.py

@@ -24,6 +24,11 @@ def goai_traduction(text, src_lang, tgt_lang):
     
     tokenizer = AutoTokenizer.from_pretrained(model_id, token=auth_token, truncation=True, max_length=512)
     model = AutoModelForSeq2SeqLM.from_pretrained(model_id, token=auth_token).to(device)
+
+    print("decoder_start_token_id:", model.config.decoder_start_token_id)
+    print("forced_bos_token_id:", model.config.forced_bos_token_id)
+    print("eos_token_id:", model.config.eos_token_id)
+
     
     # Ajout du code de langue source
     tokenizer.src_lang = src_lang