Reaper-1 / README.md
XuanzhiRen's picture
Fix collection link in model card
b94c879 verified
|
Raw History Blame Contribute Delete
2.54 kB
metadata
library_name: transformers
pipeline_tag: text-generation
base_model: FrenzyMath/Reaper-SFT
base_model_relation: finetune
tags:
  - qwen3
  - reaper
  - policy-value
  - dual-head

Reaper-1

Reaper-1 is the final policy-value checkpoint in the Reaper model family. The repository stores a Qwen3 1.7B policy model together with a separate categorical value head.

This checkpoint is part of the Reaper model collection.

Checkpoint structure

  • policy_model/: Qwen3ForCausalLM policy checkpoint
  • value_head.pt: state dictionary for a linear value head with 32 output bins
  • dual_head_meta.json: policy/value loss weights and value-head configuration
  • Tokenizer and chat-template files are stored at the repository root

The included metadata specifies policy_loss_weight=1.0, value_loss_weight=0.1, and value_dropout=0.1.

Model family

Checkpoint Stage
FrenzyMath/Reaper-Base Base checkpoint
FrenzyMath/Reaper-SFT Supervised fine-tuning
FrenzyMath/Reaper-1 Final policy-value checkpoint

Load the policy model

from transformers import AutoModelForCausalLM, AutoTokenizer

repo_id = "FrenzyMath/Reaper-1"

tokenizer = AutoTokenizer.from_pretrained(repo_id)
policy_model = AutoModelForCausalLM.from_pretrained(
    repo_id,
    subfolder="policy_model",
    torch_dtype="auto",
    device_map="auto",
)

Load the value head

import json
import torch
from huggingface_hub import hf_hub_download

meta_path = hf_hub_download(repo_id, "dual_head_meta.json")
head_path = hf_hub_download(repo_id, "value_head.pt")

with open(meta_path, encoding="utf-8") as handle:
    metadata = json.load(handle)

value_head = torch.nn.Linear(
    policy_model.config.hidden_size,
    metadata["num_value_bins"],
)
value_head.load_state_dict(
    torch.load(head_path, map_location="cpu", weights_only=True)
)
value_head.eval()

The policy and value head are stored separately. Integrating value predictions with policy hidden states depends on the calling application.

Intended use and limitations

This checkpoint is intended for research with policy-value modeling. Users should independently evaluate it for their target domain. Training-data details, benchmark results, and a license declaration are not included in the current release.