Maxvyr Maxvyr

💻

Solomaker

👨‍💻 Solomaker & Software Engineer. I build in public and share my knowledge. Creating my freedom 🏝️ one line of code at a time.

Maxvyr / grpo_demo.py

Created February 1, 2025 10:44 — forked from willccbb/grpo_demo.py

GRPO Llama-1B

	# train_grpo.py
	import re
	import torch
	from datasets import load_dataset, Dataset
	from transformers import AutoTokenizer, AutoModelForCausalLM
	from peft import LoraConfig
	from trl import GRPOConfig, GRPOTrainer

	# Load and prep dataset