Views
No views yet
1conda create -n pasodoble python=3.10.16
2conda activate pasodoble
3
4git clone https://github.com/PasoDoble-Cornell/PasoDoble.git
5cd PasoDoble
6pip install -r requirements.txt
7
8# Install flash-attention separately
9wget https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
10pip install flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
11
12# (Optional) If your current binutils version is lower than 2.38, upgrade with
13conda install -c conda-forge binutils=2.40
14
15mkdir history_recordtransformers library for text generation. Below is an example using the PasoDoble-Cornell/Qwen2.5-3b-solver-online model. Remember to replace model_id with the specific model checkpoint you intend to use.1from transformers import AutoTokenizer, AutoModelForCausalLM
2import torch
3
4model_id = "PasoDoble-Cornell/Qwen2.5-3b-solver-online" # Example model
5tokenizer = AutoTokenizer.from_pretrained(model_id)
6model = AutoModelForCausalLM.from_pretrained(
7 model_id,
8 torch_dtype=torch.bfloat16,
9 device_map="auto"
10)
11
12messages = [
13 {"role": "user", "content": "What is the capital of France?"},
14]
15
16text = tokenizer.apply_chat_template(
17 messages,
18 tokenize=False,
19 add_generation_prompt=True
20)
21model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
22
23generated_ids = model.generate(
24 model_inputs.input_ids,
25 max_new_tokens=50,
26 temperature=0.7,
27 do_sample=True,
28 eos_token_id=tokenizer.eos_token_id
29)
30generated_ids = [
31 output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
32]
33
34response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
35print(response)| Model | Training | Download |
|---|---|---|
| PasoDoble Qwen2.5-0.5B | online | 🤗 HuggingFace |
| PasoDoble Qwen2.5-0.5B | offline | 🤗 HuggingFace |
| PasoDoble Qwen2.5-1.5B | online | 🤗 HuggingFace |
| PasoDoble Qwen2.5-1.5B | offline | 🤗 HuggingFace |
| PasoDoble Qwen2.5-3B | online | 🤗 HuggingFace |
| PasoDoble Qwen2.5-3B | offline | 🤗 HuggingFace |
| PasoDoble Qwen3-0.6B | online | 🤗 HuggingFace |
| PasoDoble Qwen3-0.6B | offline | 🤗 HuggingFace |
| PasoDoble Qwen3-1.7B | online | 🤗 HuggingFace |
| PasoDoble Qwen3-1.7B | offline | 🤗 HuggingFace |
| PasoDoble Qwen3-4B | online | 🤗 HuggingFace |
| PasoDoble Qwen3-4B | offline | 🤗 HuggingFace |
1@article{zhang2025pasodoble,
2 title={Better LLM Reasoning via Dual-Play},
3 author={Zhengxin Zhang and Chengyu Huang and Aochong Oliver Li and Claire Cardie},
4 eprint={2511.11881},
5 archivePrefix={arXiv},
6 year={2025},
7 url={https://arxiv.org/abs/2511.11881}
8}