Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3
4base_id = "microsoft/Phi-3-mini-4k-instruct"
5adapter_id = "TUO_USERNAME/phi3-mr-lora-fixed-v3"
6
7tokenizer = AutoTokenizer.from_pretrained(base_id)
8
9base_model = AutoModelForCausalLM.from_pretrained(
10 base_id,
11 load_in_4bit=True,
12 device_map="auto",
13 torch_dtype="auto",
14)
15
16model = PeftModel.from_pretrained(base_model, adapter_id, device_map="auto")