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

Technical specifications

Parameters1.5B
Librarytransformers
Languagesen
Licenseqwen