[Bugfix] Fix Phi3.5 mini and MoE LoRA inference (#8571)
This commit is contained in:
17
vllm/model_executor/models/phi3.py
Normal file
17
vllm/model_executor/models/phi3.py
Normal file
@@ -0,0 +1,17 @@
|
||||
# coding=utf-8
|
||||
# Adapted from llama.py
|
||||
"""Inference-only Phi3 model code inherit from Llama.py"""
|
||||
|
||||
from vllm.model_executor.models.llama import LlamaForCausalLM
|
||||
|
||||
|
||||
class Phi3ForCausalLM(LlamaForCausalLM):
|
||||
|
||||
packed_modules_mapping = {
|
||||
"qkv_proj": [
|
||||
"qkv_proj",
|
||||
],
|
||||
"gate_up_proj": [
|
||||
"gate_up_proj",
|
||||
],
|
||||
}
|
||||
Reference in New Issue
Block a user