This model is a fine-tuned version of
alphadl/R1-Distill-0.6B-Qwen on the
open-r1/OpenR1-Math-220k dataset.
It has been trained using
TRL.
1from transformers import pipeline
2
3question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
4generator = pipeline("text-generation", model="alphadl/R1-Distill-0.6B-Qwen-GRPO", device="cuda")
5output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
6print(output["generated_text"])
