How to use aalst/starcoder2-3b-peft-lora with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("bigcode/starcoder2-3b") model = PeftModel.from_pretrained(base_model, "aalst/starcoder2-3b-peft-lora")
I also tried to fine-tune StarCoder2 with LoRA, but I was prompted to specify target_modules. What should I specify here?
· Sign up or log in to comment
