summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xconvert.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/convert.py b/convert.py
index e38ee531..980e6fc7 100755
--- a/convert.py
+++ b/convert.py
@@ -348,7 +348,7 @@ class Params:
f_rope_freq_base = 1e6
return Params(
- n_vocab=config.get("vocab_size", model["tok_embeddings.weight"].shape[0]),
+ n_vocab=model["tok_embeddings.weight"].shape[0],
n_embd=config["dim"],
n_layer=config["n_layers"],
n_ctx=n_ctx,