Update goai_helpers/goai_traduction.py

goai_helpers/goai_traduction.py

@@ -24,6 +24,8 @@ 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(model.lm_head.weight.shape)  # doit être [vocab_size, hidden_size]
+    print(model.model.shared.weight.shape)  # idem
 
     tgt_lang_id = tokenizer.convert_tokens_to_ids(tgt_lang)