Tejus tejzpr

tejzpr / grpo_demo.py

Created February 8, 2025 05:53 — 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

tejzpr / server.go

Created July 18, 2022 22:51 — forked from xcsrz/server.go

A golang web server on a randomly (os) chosen port.

	package main

	import (
	"fmt"
	"github.com/skratchdot/open-golang/open"
	"net"
	"net/http"
	)

	func main() {