diff --git a/tools/model_convert/model_convert_mt_v2.py b/tools/model_convert/model_convert_mt_v2.py index c09a905e560df8a566c5531fe129d1412fc15b96..df3fde7d1f4c0aa6a6ced11df270eb2f7e581580 100644 --- a/tools/model_convert/model_convert_mt_v2.py +++ b/tools/model_convert/model_convert_mt_v2.py @@ -75,7 +75,7 @@ class ModelConverter: transformed_data = dict(zip(result_key[:], result_embedding[:])) save_path = os.path.join(self._output_path, table_name + "_key_embedding" + ".npy") save_dir = os.path.dirname(save_path) - os.makedis(save_dir, exist_ok=True) + os.makedirs(save_dir, exist_ok=True) np.save(save_path, transformed_data) def _get_key_and_offset(self, sparse_file_path, table_name):