How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("image-text-to-text", model="two-tiger/VRPRM-rl")
messages = [
    {
        "role": "user",
        "content": [
            {"type": "image", "url": "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/p-blog/candy.JPG"},
            {"type": "text", "text": "What animal is on the candy?"}
        ]
    },
]
pipe(text=messages)
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoProcessor, AutoModelForMultimodalLM

processor = AutoProcessor.from_pretrained("two-tiger/VRPRM-rl")
model = AutoModelForMultimodalLM.from_pretrained("two-tiger/VRPRM-rl", device_map="auto")
messages = [
    {
        "role": "user",
        "content": [
            {"type": "image", "url": "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/p-blog/candy.JPG"},
            {"type": "text", "text": "What animal is on the candy?"}
        ]
    },
]
inputs = processor.apply_chat_template(
	messages,
	add_generation_prompt=True,
	tokenize=True,
	return_dict=True,
	return_tensors="pt",
).to(model.device)

outputs = model.generate(**inputs, max_new_tokens=256)
print(processor.decode(outputs[0][inputs["input_ids"].shape[-1]:]))
Quick Links

VRPRM-rl

VRPRM-rl is the merged process-supervised RL checkpoint released for VRPRM. It is initialized from VRPRM-sft and trained with GSPO-token/GRPO LoRA RL on 40K confident VisualPRM400K-derived examples.

Training

  • Initialization: VRPRM-sft
  • Data: 40K train / 2K validation
  • Rollouts: 16 per prompt, temperature 0.8, top-p 0.95
  • Reward: 0.92 * step + 0.05 * format + 0.03 * think
  • Optimization: lr 1e-6, KL coefficient 0.2, 600 steps
  • Public artifact: end-of-run merged checkpoint

Evaluation

VisualProcessBench official no-reference protocol:

Model Overall step macro-F1
VRPRM-rl 69.18

The public artifact is the end-of-run checkpoint. The run's highest RL validation reward occurred earlier; comparisons should use the exact released artifact and protocol documented in the code repository.

Usage

Use the same global-thinking stepwise protocol as VRPRM-sft. The model is evaluated with benchmarks/visualprocessbench/eval_vpb_global_think.py.

Limitations

This research model is not a general assistant and is not suitable for high-stakes automated grading without human oversight.

Downloads last month
27
Safetensors
Model size
9B params
Tensor type
BF16
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Model tree for two-tiger/VRPRM-rl

Finetuned
(80)
this model

Collection including two-tiger/VRPRM-rl