Views
No views yet
1from peft import PeftModel
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4base = AutoModelForCausalLM.from_pretrained(
5 "kshitijthakkar/deepseek-v4-mini-300M-from-flash", trust_remote_code=True
6)
7model = PeftModel.from_pretrained(base, "kshitijthakkar/deepseek-v4-mini-300M-from-flash-sft-test-lora")
8tokenizer = AutoTokenizer.from_pretrained("kshitijthakkar/deepseek-v4-mini-300M-from-flash", trust_remote_code=True)