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