How to use CMU-AIR2/deepseek-math-base-LORA-ArithSteps-6k with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("deepseek-ai/deepseek-math-7b-base") model = PeftModel.from_pretrained(base_model, "CMU-AIR2/deepseek-math-base-LORA-ArithSteps-6k")
