Views
No views yet
1from peft import PeftModel
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4# ベースモデルのロード
5model = AutoModelForCausalLM.from_pretrained("llm-jp/llm-jp-3-13b")
6tokenizer = AutoTokenizer.from_pretrained("llm-jp/llm-jp-3-13b")
7
8# LoRAの適用
9model = PeftModel.from_pretrained(model, "path/to/lora/weights")