Views
No views yet
1from peft import PeftModel
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4base = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-3.2-1B")
5model = PeftModel.from_pretrained(base, "arvindcr4/llama-3.2-1b-distillation-offpolicy-lora")
6tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-1B")