Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained(
4 "KhushalM/Qwen2.5-1.5B-GRPO-Complete",
5 torch_dtype="auto",
6 trust_remote_code=True
7)
8
9tokenizer = AutoTokenizer.from_pretrained(
10 "KhushalM/Qwen2.5-1.5B-GRPO-Complete",
11 trust_remote_code=True
12)