Skip to content

Instantly share code, notes, and snippets.

View ehabhussein's full-sized avatar
🎯
Focusing

Ehab Hussein ehabhussein

🎯
Focusing
View GitHub Profile
@ehabhussein
ehabhussein / grpo_demo.py
Created February 2, 2025 18:37 — 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
#!/usr/bin/python
#Nuc220LE3AN Ida python script to create system memory map segments
from idautils import *
from idc import *
from idaapi import *
SEGMENT_ARCH = 32
for i in [