fix convert.py for codellama, add llama 34B to the list of recognized models (#2768)

1 files changed, 1 insertions, 1 deletions

diff --git a/convert.py b/convert.py
index 8d34d5f2..d3776a4d 100755
--- a/convert.py
+++ b/convert.py
@@ -191,7 +191,7 @@ class Params:
     def loadOriginalParamsJson(model: 'LazyModel', config_path: 'Path') -> 'Params':
         config = json.load(open(config_path))
 
-        n_vocab    = config["vocab_size"]
+        n_vocab    = config["vocab_size"] if "vocab_size" in config else -1
         n_embd     = config["dim"]
         n_layer    = config["n_layers"]
         n_mult     = config["multiple_of"]