Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_id = "david-thrower/HelixLM-384d-cca-43M-50Kt-medium"
4tokenizer = AutoTokenizer.from_pretrained(model_id)
5model = AutoModelForCausalLM.from_pretrained(model_id)AutoModelForCausalLM with the appropriate AutoModel class.