Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_id = 'aerialblancaservices/v2rmp-agent-1.5b-merged'
4tokenizer = AutoTokenizer.from_pretrained(model_id)
5model = AutoModelForCausalLM.from_pretrained(model_id)AutoModelForCausalLM with the appropriate AutoModel class.