Xcvddax/Attack-techniques e8869ce verified
Chihhh Linnn commited on
How to use chihhh/Attack-techniques-Lora-gemma with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("mustafaaljadery/gemma-2B-10M")
model = PeftModel.from_pretrained(base_model, "chihhh/Attack-techniques-Lora-gemma")