Views
No views yet
1pip install -U transformers>=4.48.0
2pip install flash-attn1from transformers import AutoTokenizer, AutoModel
2device = 'cuda:0'
3
4model_name = "manelalab/chrono-bert-v1-19991231"
5
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7model = AutoModel.from_pretrained(model_name).to(device)
8
9text = "Obviously, the time continuum has been disrupted, creating a new temporal event sequence resulting in this alternate reality. -- Dr. Brown, Back to the Future Part II"
10
11inputs = tokenizer(text, return_tensors="pt").to(device)
12outputs = model(**inputs)1from transformers import AutoTokenizer, AutoModelForMaskedLM
2device = 'cuda:0'
3
4model_name = "manelalab/chrono-bert-v1-20201231"
5
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7model = AutoModelForMaskedLM.from_pretrained(model_name).to(device)
8
9year_election = 2016
10year_begin = year_election+1
11text = f"After the {year_election} U.S. presidential election, President [MASK] was inaugurated as U.S. President in the year {year_begin}."
12
13inputs = tokenizer(text, return_tensors="pt").to(device)
14outputs = model(**inputs)
15masked_index = inputs["input_ids"][0].tolist().index(tokenizer.mask_token_id)
16predicted_token_id = outputs.logits[0, masked_index].argmax(axis=-1)
17predicted_token = tokenizer.decode(predicted_token_id)@article{He2025ChronoBERT,
title={Chronologically Consistent Large Language Models},
author={He, Songrun and Lv, Linying and Manela, Asaf and Wu, Jimmy},
journal={Working Paper},
year={2025}
}