convert : fix n_ff typo

This commit is contained in:
Georgi Gerganov 2023-12-09 10:59:37 +02:00
parent dff8cbeb39
commit d38e41ee69
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735

View File

@ -275,7 +275,7 @@ class Params:
n_ff = model["layers.0.feed_forward.experts.0.w1.weight"].shape[0]
n_ctx = 32768
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(
n_vocab = model["tok_embeddings.weight"].shape[0],