convert : fix n_ff typo
This commit is contained in:
parent
dff8cbeb39
commit
d38e41ee69
1 changed files with 1 additions and 1 deletions
|
@ -275,7 +275,7 @@ class Params:
|
||||||
n_ff = model["layers.0.feed_forward.experts.0.w1.weight"].shape[0]
|
n_ff = model["layers.0.feed_forward.experts.0.w1.weight"].shape[0]
|
||||||
n_ctx = 32768
|
n_ctx = 32768
|
||||||
else:
|
else:
|
||||||
n_ff = model["layers.0.feed_forward.w1.weight"].shape[0],
|
n_ff = model["layers.0.feed_forward.w1.weight"].shape[0]
|
||||||
|
|
||||||
return Params(
|
return Params(
|
||||||
n_vocab = model["tok_embeddings.weight"].shape[0],
|
n_vocab = model["tok_embeddings.weight"].shape[0],
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue