diff options
author | Suaj Carrot <72162667+SuajCarrot@users.noreply.github.com> | 2023-03-19 12:38:44 -0600 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-03-19 20:38:44 +0200 |
commit | 7392f1cd2cef4dfed41f4db7c4160ab86c0dfcd9 (patch) | |
tree | b5274321bf1861ac36108b4ae722a3c003390b4c /.dockerignore | |
parent | ad5fd5b60cfdfbfb22b0f2bc9e9f6c9692768f8d (diff) |
Improved quantize script (#222)
* Improved quantize script
I improved the quantize script by adding error handling and allowing to select many models for quantization at once in the command line. I also converted it to Python for generalization as well as extensibility.
* Fixes and improvements based on Matt's observations
Fixed and improved many things in the script based on the reviews made by @mattsta. The parallelization suggestion is still to be revised, but code for it was still added (commented).
* Small fixes to the previous commit
* Corrected to use the original glob pattern
The original Bash script uses a glob pattern to match files that have endings such as ...bin.0, ...bin.1, etc. That has been translated correctly to Python now.
* Added support for Windows and updated README to use this script
New code to set the name of the quantize script binary depending on the platform has been added (quantize.exe if working on Windows) and the README.md file has been updated to use this script instead of the Bash one.
* 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.
* Corrected previous commit
* Small tweak: changed the name of the program in argparse
This was making the automatic help message to be suggesting the program's usage as being literally "$ Quantization Script [arguments]". It should now be something like "$ python3 quantize.py [arguments]".
Diffstat (limited to '.dockerignore')
0 files changed, 0 insertions, 0 deletions