vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

model: dbrx: fix expert reshape

Browse source
This commit is contained in:
Pierrick HYMBERT 2024-04-07 19:12:24 +02:00
parent dbfd59114f
commit 7dd84b0924
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
convert-hf-to-gguf.py
View file

@ -1548,7 +1548,7 @@ class DbrxModel(Model):
# Reshape experts tensors from 2D to 3D as expected by GeLU
if experts and n_dims == 2:
data = data.reshape((self.hparams["d_model"], self.hparams["ffn_config"]["ffn_hidden_size"], self.hparams["ffn_config"]["moe_num_experts"]))
data = data.reshape((self.hparams["ffn_config"]["moe_num_experts"], self.hparams["ffn_config"]["ffn_hidden_size"], self.hparams["d_model"]))
n_dims = len(data.shape)
# if f32 desired, convert any float16 to float32