diff options
| author | Guillaume Klein <guillaumekln@users.noreply.github.com> | 2021-10-21 13:24:18 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-10-21 13:24:18 +0300 |
| commit | 97d93de80177d70387283d69fa8b4a45bd7b1454 (patch) | |
| tree | 2bea6d518fb07d6f16cf149a111a553bc5947a51 | |
| parent | e2628a2192b39a25606b1638ea2b2a38f21306e5 (diff) | |
Add int8_float16 as possible quantization value for CTranslate2 (#2116)
| -rwxr-xr-x | onmt/bin/release_model.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/onmt/bin/release_model.py b/onmt/bin/release_model.py index 9c9a67a5..f12323e3 100755 --- a/onmt/bin/release_model.py +++ b/onmt/bin/release_model.py @@ -15,7 +15,7 @@ def main(): default="pytorch", help="The format of the released model") parser.add_argument("--quantization", "-q", - choices=["int8", "int16", "float16"], + choices=["int8", "int16", "float16", "int8_float16"], default=None, help="Quantization type for CT2 model.") opt = parser.parse_args() |