Commit
•
7128af0
1
Parent(s):
aa18a06
Update README.md
Browse files
README.md
CHANGED
@@ -86,8 +86,8 @@ target_lang = "frp"
|
|
86 |
max_input_length = 128
|
87 |
max_target_length = 128
|
88 |
|
89 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
90 |
-
model = AutoModelForSeq2SeqLM.from_pretrained(
|
91 |
|
92 |
inputs = tokenizer("Je mange une pomme", return_tensors="pt").input_ids
|
93 |
outputs = model.generate(inputs.to("cuda:0"), max_new_tokens=40, do_sample=True, top_k=30, top_p=0.95)
|
|
|
86 |
max_input_length = 128
|
87 |
max_target_length = 128
|
88 |
|
89 |
+
tokenizer = AutoTokenizer.from_pretrained("Propicto/t2p-t5-large-orfeo")
|
90 |
+
model = AutoModelForSeq2SeqLM.from_pretrained("Propicto/t2p-t5-large-orfeo")
|
91 |
|
92 |
inputs = tokenizer("Je mange une pomme", return_tensors="pt").input_ids
|
93 |
outputs = model.generate(inputs.to("cuda:0"), max_new_tokens=40, do_sample=True, top_k=30, top_p=0.95)
|