Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_id = "zyberg2091/fineweb_pretrained_model_37M"
4
5tokenizer = AutoTokenizer.from_pretrained(model_id)
6model = AutoModelForCausalLM.from_pretrained(
7 model_id,
8 trust_remote_code=True,
9)