ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

fix phi3 (#11729)

Browse source
This commit is contained in:
Yishuo Wang 2024-08-07 09:39:46 +08:00 committed by GitHub
parent e32d13d78c
commit c093f7d980
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
python/llm/src/ipex_llm/transformers/models/phi3.py
View file

@ -61,7 +61,8 @@ def pre_compute_inv_freq(module: torch.nn.Module):
module.base **
(torch.arange(0, module.dim, 2, dtype=torch.int64).float() / module.dim)
)
else:
elif module.__class__.__name__ in ["Phi3SuScaledRotaryEmbedding",
"Phi3LongRoPEScaledRotaryEmbedding"]:
inv_freq_shape = torch.arange(0, module.dim, 2, dtype=torch.int64).float() / module.dim
short_ext_factors = torch.tensor(module.short_factor, dtype=torch.float32)
module.inv_freq = 1.0 / (short_ext_factors * module.base ** inv_freq_shape)