fblgit / miniclaus-qw1.5B-UNAMGS-GRPO
Text Generation1.5B765Updated Feb 6, 2025
Downloads
76
Likes
5
Last updated
Feb 6, 2025
License
qwen
This version is RL with GRPO on GSM8k for 1400 steps using this code: # traingrpo.py import re import torch from datasets import loaddataset, Dataset from transformers import AutoTokenizer, AutoModelForCausalLM from peft import LoraConfig from trl import GRPOConfig, GRPOTrainer
Tasks & library
- text generation
- transformers
Tags
- transformers
- qwen2
- text-generation
- generated_from_trainer
- conversational
- en
