Skip to content
Closed
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions vllm/model_executor/models/phi_1_5.py
Original file line number Diff line number Diff line change
Expand Up @@ -183,7 +183,7 @@ def __init__(self,
linear_method: Optional[LinearMethodBase] = None):
super().__init__()
self.ln = nn.LayerNorm(config.hidden_size,
eps=config.layer_norm_epsilon)
eps=config.layer_norm_eps)
self.mixer = PhiAttention(config, linear_method)
self.mlp = PhiMLP(config, linear_method)

Expand Down Expand Up @@ -245,7 +245,7 @@ class PhiCausalLMHead(nn.Module):
def __init__(self, config: PretrainedConfig):
super().__init__()
self.ln = nn.LayerNorm(config.hidden_size,
eps=config.layer_norm_epsilon)
eps=config.layer_norm_eps)
self.linear = ParallelLMHead(config.vocab_size,
config.hidden_size,
bias=True)
Expand Down