Check the existence of f16_model_path_base in quantize.py (#574)

Co-authored-by: Jincheng Miao <jincheng.miao@gmail.com>
This commit is contained in:
dotpy314 2023-03-28 23:06:28 +08:00 committed by GitHub
parent a6bdc47cba
commit 28ba975aea
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -74,6 +74,10 @@ def main():
args.models_path, model, "ggml-model-f16.bin" args.models_path, model, "ggml-model-f16.bin"
) )
if not os.path.isfile(f16_model_path_base):
print(f'The file %s was not found' % f16_model_path_base)
sys.exit(1)
f16_model_parts_paths = map( f16_model_parts_paths = map(
lambda filename: os.path.join(f16_model_path_base, filename), lambda filename: os.path.join(f16_model_path_base, filename),
glob.glob(f"{f16_model_path_base}*") glob.glob(f"{f16_model_path_base}*")