Views
No views yet
1from peft import PeftModel
2from transformers import AutoModelForCausalLM, AutoTokenizer
3base = "unsloth/Qwen2.5-3B-bnb-4bit"
4tok = AutoTokenizer.from_pretrained(base)
5m = AutoModelForCausalLM.from_pretrained(base, load_in_4bit=True, device_map="auto")
6m = PeftModel.from_pretrained(m, "TunaMonsieur/lab22-dpo-vn")