Views
No views yet
1from peft import PeftModel
2from transformers import AutoModelForCausalLM
3
4base = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2.5-1.5B", torch_dtype="bfloat16")
5model = PeftModel.from_pretrained(base, "sandywong/distractor-lora-clean-only-qwen25-1.5b")
6
7# Or merge into base for inference:
8model = model.merge_and_unload()