AI-MO/NuminaMath-CoT
Viewer • Updated • 860k • 56.9k • 582
How to use aphoticshaman/elle-72b-math-v2 with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2.5-72B-Instruct")
model = PeftModel.from_pretrained(base_model, "aphoticshaman/elle-72b-math-v2")Elle-72B-Math-v2 is a LoRA adapter fine-tuned on Qwen/Qwen2.5-72B-Instruct for mathematical reasoning using the NuminaMath-CoT dataset.
Fine-tuned using:
from peft import PeftModel
from transformers import AutoModelForCausalLM, AutoTokenizer
base_model = AutoModelForCausalLM.from_pretrained(
"Qwen/Qwen2.5-72B-Instruct",
torch_dtype="auto",
device_map="auto",
trust_remote_code=True
)
model = PeftModel.from_pretrained(base_model, "aphoticshaman/elle-72b-math-v2")
tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-72B-Instruct")
Apache 2.0