Views
No views yet
1
2import torch
3from peft import PeftModel
4from transformers import AutoProcessor, AutoModelForCausalLM
5
6def create_model(model_name_or_path = 'microsoft/Phi-4-multimodal-instruct'):
7 model = AutoModelForCausalLM.from_pretrained(
8 model_name_or_path,
9 device_map="auto",
10 trust_remote_code=True,
11 cache_dir = os.getenv('CACHE_DIR'),
12 _attn_implementation='sdpa',
13 torch_dtype = torch.bfloat16
14 )
15 return model
16
17model_name_or_path = 'microsoft/Phi-4-multimodal-instruct'
18model = create_model(model_name_or_path)
19
20model.load_adapter("binhquoc/alm-add-phi4-non", adapter_name="speech")
21
22model.set_adapter("speech")
23processor = AutoProcessor.from_pretrained(
24 model_name_or_path,
25 trust_remote_code=True,
26 cache_dir = os.getenv('CACHE_DIR'),
27)