Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3
4# 토크나이저 로드
5tokenizer = AutoTokenizer.from_pretrained("merged_model_new_tokenizer")
6
7# 베이스 모델 로드
8base_model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-3.2-1B-Instruct")
9
10# LoRA 모델 로드
11model = PeftModel.from_pretrained(base_model, "merged_model_new_tokenizer")