ArissBandoss commited on
Commit
ca5e581
·
verified ·
1 Parent(s): d4b611c

Update goai_helpers/goai_traduction.py

Browse files
Files changed (1) hide show
  1. goai_helpers/goai_traduction.py +3 -1
goai_helpers/goai_traduction.py CHANGED
@@ -24,7 +24,9 @@ def goai_traduction(text, src_lang, tgt_lang):
24
 
25
  tokenizer = AutoTokenizer.from_pretrained(model_id, token=auth_token)
26
  model = AutoModelForSeq2SeqLM.from_pretrained(model_id, token=auth_token).to(device)
27
-
 
 
28
  # Configuration du tokenizer
29
  tokenizer.src_lang = src_lang
30
 
 
24
 
25
  tokenizer = AutoTokenizer.from_pretrained(model_id, token=auth_token)
26
  model = AutoModelForSeq2SeqLM.from_pretrained(model_id, token=auth_token).to(device)
27
+
28
+ print(tokenizer.model_max_length)
29
+ print(model.model.encoder.embed_positions.weight.shape)
30
  # Configuration du tokenizer
31
  tokenizer.src_lang = src_lang
32