Skip to content

Commit 35ce867

Browse files
authored
Merge branch 'main' into hf-flagembedding-download
2 parents 0552235 + f87330f commit 35ce867

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

fastembed/embedding.py

+8-3
Original file line numberDiff line numberDiff line change
@@ -62,9 +62,14 @@ def load_tokenizer(cls, model_dir: Path, max_length: int = 512) -> Tokenizer:
6262
if not tokens_map_path.exists():
6363
raise ValueError(f"Could not find special_tokens_map.json in {model_dir}")
6464

65-
config = json.load(open(str(config_path)))
66-
tokenizer_config = json.load(open(str(tokenizer_config_path)))
67-
tokens_map = json.load(open(str(tokens_map_path)))
65+
with open(str(config_path)) as config_file:
66+
config = json.load(config_file)
67+
68+
with open(str(tokenizer_config_path)) as tokenizer_config_file:
69+
tokenizer_config = json.load(tokenizer_config_file)
70+
71+
with open(str(tokens_map_path)) as tokens_map_file:
72+
tokens_map = json.load(tokens_map_file)
6873

6974
tokenizer = Tokenizer.from_file(str(tokenizer_path))
7075
tokenizer.enable_truncation(max_length=min(tokenizer_config["model_max_length"], max_length))

0 commit comments

Comments
 (0)