Views
No views yet
models/gemma-2-9b-it-taboo-leaf-merged1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3
4# Load base model and tokenizer
5base_model = AutoModelForCausalLM.from_pretrained("models/gemma-2-9b-it-taboo-leaf-merged")
6tokenizer = AutoTokenizer.from_pretrained("models/gemma-2-9b-it-taboo-leaf-merged")
7
8# Load LoRA adapter
9model = PeftModel.from_pretrained(base_model, "model-organisms-for-real/gemma2_9b_it_taboo_leaf_oracle_v1")