How to use ktm379/code-llama-7b-train with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("TinyPixel/Llama-2-7B-bf16-sharded") model = PeftModel.from_pretrained(base_model, "ktm379/code-llama-7b-train")
56a5425
1
2
3
4
5
{ "<|im_end|>": 32000, "<|im_start|>": 32001 }
