How to use mtzig/v3_mistral_balance1_base_lora with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("peiyi9979/math-shepherd-mistral-7b-prm") model = PeftModel.from_pretrained(base_model, "mtzig/v3_mistral_balance1_base_lora")
This model has 2 files scanned as unsafe.
How to fix it?
What is a pickle import?
