OpenMOSS-Team/moss-003-sft-data
Preview • Updated • 719 • 68
How to use Laurie/baichuan-7b-qlora-moss with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("baichuan-inc/baichuan-7B")
model = PeftModel.from_pretrained(base_model, "Laurie/baichuan-7b-qlora-moss")The following bitsandbytes quantization config was used during training:
git clone https://huggingface.co/Laurie/baichuan-7b-qlora-moss
cd baichuan-7b-qlora-moss
python src/web_demo.py \
--model_name_or_path baichuan-inc/baichuan-7B \
--checkpoint_dir .