nitinvenu/SG_TestDataSet
Viewer • Updated • 1.22k • 4
How to use nitinvenu/SG_LLama_1 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, "nitinvenu/SG_LLama_1")