From b802b788c42f16d1371a8a03387085c1207b7b51 Mon Sep 17 00:00:00 2001 From: Gerardo Romero Date: Sun, 19 Mar 2023 12:15:47 -0600 Subject: [PATCH] Fixed a typo and removed shell=True in the subprocess.run call Fixed a typo regarding the new filenames of the quantized models and removed the shell=True parameter in the subprocess.run call as it was conflicting with the list of parameters. --- quantize.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/quantize.py b/quantize.py index 97d841777..8ac0d58a8 100644 --- a/quantize.py +++ b/quantize.py @@ -104,7 +104,7 @@ def __run_quantize_script(script_path, f16_model_part_path): f16 model to quantize. """ - new_quantized_model_path = f16_model_part_path.replace("16", "q4_0") + new_quantized_model_path = f16_model_part_path.replace("f16", "q4_0") subprocess.run( [script_path, f16_model_part_path, new_quantized_model_path, "2"], shell=True, check=True