Просмотр исходного кода

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

Co-authored-by: Jincheng Miao <jincheng.miao@gmail.com>
dotpy314 2 лет назад
Родитель
Сommit
28ba975aea
1 измененных файлов с 4 добавлено и 0 удалено
  1. 4 0
      quantize.py

+ 4 - 0
quantize.py

@@ -74,6 +74,10 @@ def main():
             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(
             lambda filename: os.path.join(f16_model_path_base, filename),
             glob.glob(f"{f16_model_path_base}*")