Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "ambrosfitz/qwen-1.5b-book-rarity-grpo"
4tokenizer = AutoTokenizer.from_pretrained(model_name)
5model = AutoModelForCausalLM.from_pretrained(model_name)
6
7prompt = """Analyze this book for rarity and marketplace value. Provide step-by-step reasoning.
8
9Title: First Edition Book Title
10Author: Author Name
11Year: 1990
12Holdings: 5 libraries
13Tier: 2
14Thesis: 0
15Gov Doc: 0
16
17Think through: holdings, language, document type, age, and rarity tier."""
18
19inputs = tokenizer(prompt, return_tensors="pt")
20outputs = model.generate(**inputs, max_new_tokens=256)
21print(tokenizer.decode(outputs[0], skip_special_tokens=True))| Metric | Start | Final | Improvement |
|---|---|---|---|
| Reward | 1.86 | 3.17 | +67% |
| Reward Std | 0.99 | 0.53 | -46% (more stable) |
| KL Divergence | 0.001 | 0.013 | Controlled |