Views
No views yet
python train.py1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3
4# Load the trained model
5base_model = AutoModelForCausalLM.from_pretrained("google/gemma-2-2b")
6model = PeftModel.from_pretrained(base_model, "Kronu/gemma-2-2b-lean-expert-optimized")
7tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-2b")