Views
No views yet
google/gemma-2-9b-it1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3
4# Load base model and tokenizer
5base_model = AutoModelForCausalLM.from_pretrained("google/gemma-2-9b-it")
6tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-9b-it")
7
8# Load LoRA adapter
9model = PeftModel.from_pretrained(base_model, "thejaminator/gemma-hook-layer-0")