Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_id = 'Mohit-0308/transaction-ner-bank'
4tokenizer = AutoTokenizer.from_pretrained(model_id)
5model = AutoModelForCausalLM.from_pretrained(model_id)AutoModelForCausalLM with the appropriate AutoModel class.