Views
No views yet
1from transformers import AutoModelForMaskedLM, AutoTokenizer
2
3repo_id = "WesScivetti/GPT-BERT_NtoN_50Zipf_Seed3"
4tokenizer = AutoTokenizer.from_pretrained(repo_id)
5model = AutoModelForMaskedLM.from_pretrained(repo_id, trust_remote_code=True)
6model.eval()