File tree Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Original file line number Diff line number Diff line change 17
17
# and vocabulary.
18
18
#
19
19
import argparse
20
+ import os
20
21
import sys
21
22
import json
22
23
import struct
@@ -44,8 +45,14 @@ def get_n_parts(dim):
44
45
45
46
def load_hparams_and_tokenizer (dir_model ):
46
47
48
+ # `dir_model` is something like `models/7B` or `models/7B/`.
49
+ # "tokenizer.model" is expected under model's parent dir.
50
+ # When `dir_model` is a symlink, f"{dir_model}/../tokenizer.model" would not be found.
51
+ # Let's use the model's parent dir directly.
52
+ model_parent_dir = os .path .dirname (os .path .normpath (dir_model ))
53
+
47
54
fname_hparams = f"{ dir_model } /params.json"
48
- fname_tokenizer = f"{ dir_model } /.. /tokenizer.model"
55
+ fname_tokenizer = f"{ model_parent_dir } /tokenizer.model"
49
56
50
57
with open (fname_hparams , "r" ) as f :
51
58
hparams = json .load (f )
You can’t perform that action at this time.
0 commit comments