Views
No views yet
checkpoint-1000/transformers1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "accuracy-maker/Llama-3.2-1B-GRPO-gsm8k"
4
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForCausalLM.from_pretrained(model_name)
7
8input_text = "What is the capital of France?"
9generate_with_stream(input_text)
10