Views
No views yet
1from peft import PeftModel
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4base_model = AutoModelForCausalLM.from_pretrained("google/gemma-2-2b-it")
5model = PeftModel.from_pretrained(base_model, "ssmurali/gemma2-2b-safeorpo")
6tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-2b-it")