Views
No views yet
la)Note: This model operates on continuous text streams. While it restores sentence boundaries (.,?,!) and capitalization, it is not designed to predict paragraph breaks or layout features.
"punctuate: " and expects lowercased input. The helper function below handles this automatically.1import torch
2from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
4device = "cuda" if torch.cuda.is_available() else "cpu"
5
6model_name = "mschonhardt/mt5-latin-punctuator-large"
7tokenizer = AutoTokenizer.from_pretrained(model_name)
8model = AutoModelForSeq2SeqLM.from_pretrained(model_name).to(device)
9
10def punctuate(text: str) -> str:
11 # Preprocessing: Add prefix and lowercase as per training script
12 input_text = "punctuate: " + text.lower()
13
14 inputs = tokenizer(
15 input_text,
16 return_tensors="pt",
17 truncation=True,
18 max_length=1024,
19 ).to(device)
20
21 with torch.no_grad():
22 output_ids = model.generate(
23 **inputs,
24 max_length=1024,
25 num_beams=4,
26 early_stopping=True,
27 )
28 return tokenizer.decode(output_ids[0], skip_special_tokens=True)
29
30# Example usage
31text = "gallia est omnis divisa in partes tres"
32print(punctuate(text))
33# Output: Gallia est omnis divisa in partes tres..,;?!) was removed, and the prefix "punctuate: " was added.transformers Seq2SeqTrainer.bfloat16 (BF16) mixed precision| Parameter | Value |
|---|---|
| Batch Size | 4 |
| Grad Accumulation | 16 steps |
| Effective Batch Size | 64 |
| Learning Rate | 5e-5 (Linear Decay) |
| Max Sequence Length | 1024 (Source and Target) |
| Gradient Checkpointing | True |
| Epoch | Validation Loss |
|---|---|
| 0.2 | 0.1448 |
| 0.6 | 0.1137 |
| 1.0 | 0.1043 |
| 1.4 | 0.0982 |
| 1.6 | 0.0973 (Best) |
Input: gallia est omnis divisa in partes tres quarum unam incolunt belgae aliam aquitani tertiam qui ipsorum lingua celtae nostra galli appellanturOutput: Gallia est omnis divisa in partes tres, quarum unam incolunt Belgae, aliam Aquitani, tertiam qui ipsorum lingua Celtae, nostra Galli appellantur.
Input: Consilium ad conceptionem mondini bononiensis de leuciis quia ex relatis principio sterilitatis causam fore duplicem horum coniugatorum unam ocultam scilicet disproportionem eorum nobis ignotam cuius remotioni altissimus provideat quia nostrum non est nec posse alicuius artis et aliam manifestam existimo esse complectionem eorum declinare ad frigidum et declinabilem in processu etatis etiam ad frigidum et humidum humiditatem accidentali corporibus eorum est subveniendum triplici instrumento medici declinante ad calidum et siccum quo ad medicinas et ad calidum et humidum quantum ad dictamOutput: Consilium ad conceptionem Mondini Bononiensis de leuciis. Quia ex relatis principio sterilitatis causam fore duplicem horum coniugatorum, unam ocultam, scilicet disproportionem eorum nobis ignotam, cuius remotioni Altissimus provideat, quia nostrum non est nec posse alicuius artis, et aliam manifestam existimo esse complectionem eorum declinare ad frigidum et declinabilem in processu etatis etiam ad frigidum et humidum. Humiditatem accidentali corporibus eorum est subveniendum triplici instrumento medici declinante ad calidum et siccum quo ad medicinas et ad calidum et humidum quantum ad dictam.
1@misc{schonhardt-2025-latin-punctuator,
2 author = {Schonhardt, Michael},
3 title = {mT5 Latin Punctuator (mt5-large)},
4 year = {2025},
5 publisher = {Hugging Face},
6 doi = {10.5281/zenodo.17777660}
7 howpublished = {\url{[https://huggingface.co/mschonhardt/mt5-latin-punctuator-large](https://huggingface.co/mschonhardt/mt5-latin-punctuator-large)}},
8 note = {Part of the LOEWE Exploration 'Embedding the Past'. Data provided by LTA}
9}1@inproceedings{xue-etal-2021-mt5,
2 title = "m{T}5: A Massively Multilingual Pre-trained Text-to-Text Transformer",
3 author = "Xue, Linting and
4 Constant, Noah and
5 Roberts, Adam and
6 Kale, Mihir and
7 Al-Rfou, Rami and
8 Siddhant, Aditya and
9 Barua, Aditya and
10 Raffel, Colin",
11 editor = "Toutanova, Kristina and
12 Rumshisky, Anna and
13 Zettlemoyer, Luke and
14 Hakkani-Tur, Dilek and
15 Beltagy, Iz and
16 Bethard, Steven and
17 Cotterell, Ryan and
18 Chakraborty, Tanmoy and
19 Zhou, Yichao",
20 booktitle = "Proceedings of the 2021 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies",
21 month = jun,
22 year = "2021",
23 address = "Online",
24 publisher = "Association for Computational Linguistics",
25 url = "https://aclanthology.org/2021.naacl-main.41/",
26 doi = "10.18653/v1/2021.naacl-main.41",
27 pages = "483--498"
28
29}