Moore-Language-Space-ZeroGPU

Sleeping

ArissBandoss commited on May 19

Commit

b82534e

verified ·

1 Parent(s): 5a98d3d

Update goai_helpers/goai_traduction.py

Files changed (1) hide show

goai_helpers/goai_traduction.py CHANGED Viewed

@@ -26,7 +26,7 @@ def goai_traduction(text, src_lang, tgt_lang):
     model = AutoModelForSeq2SeqLM.from_pretrained(model_id, token=auth_token).to(device)
     print(tokenizer.model_max_length)
-    print(model.model.encoder.embed_positions.weight.shape)
     # Configuration du tokenizer
     tokenizer.src_lang = src_lang

     model = AutoModelForSeq2SeqLM.from_pretrained(model_id, token=auth_token).to(device)
     print(tokenizer.model_max_length)
+    print(model.model.encoder.embed_positions.weights.shape)
     # Configuration du tokenizer
     tokenizer.src_lang = src_lang