How to use SwastikM/Llama-2-7B-Chat-text2code with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("TheBloke/Llama-2-7b-Chat-GPTQ") model = PeftModel.from_pretrained(base_model, "SwastikM/Llama-2-7B-Chat-text2code")
flytech