Propicto
/

t2p-t5-large-orfeo

text2text-generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

cecilemacaire commited on Jul 5

Commit

7128af0

•

1 Parent(s): aa18a06

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -86,8 +86,8 @@ target_lang = "frp"
 max_input_length = 128
 max_target_length = 128
-tokenizer = AutoTokenizer.from_pretrained(checkpoint)
-model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint)
 inputs = tokenizer("Je mange une pomme", return_tensors="pt").input_ids
 outputs = model.generate(inputs.to("cuda:0"), max_new_tokens=40, do_sample=True, top_k=30, top_p=0.95)

 max_input_length = 128
 max_target_length = 128
+tokenizer = AutoTokenizer.from_pretrained("Propicto/t2p-t5-large-orfeo")
+model = AutoModelForSeq2SeqLM.from_pretrained("Propicto/t2p-t5-large-orfeo")
 inputs = tokenizer("Je mange une pomme", return_tensors="pt").input_ids
 outputs = model.generate(inputs.to("cuda:0"), max_new_tokens=40, do_sample=True, top_k=30, top_p=0.95)