Views
No views yet
1from peft import PeftModel
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4# 加载基座模型
5base_model = AutoModelForCausalLM.from_pretrained("GSAI-ML/LLaDA-1.5")
6tokenizer = AutoTokenizer.from_pretrained("GSAI-ML/LLaDA-1.5")
7
8# 加载 adapter
9model = PeftModel.from_pretrained(base_model, "mousezhang/countdown-llada1.5")