Small fix to enable gen_external_params.py script to not quantize (no…

…d-ai#247)
gpetters-amd · Dec 13, 2023 · 10c8ed1 · 10c8ed1
1 parent d5ae81a
commit 10c8ed1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/turbine_models/gen_external_params/gen_external_params.py b/python/turbine_models/gen_external_params/gen_external_params.py
@@ -19,7 +19,7 @@
     "--quantization",
     type=str,
     default="int4",
-    choices=["int4", "int8"],
+    choices=["unquantized", "int4", "int8"],
     help="Type of quantization to apply.",
 )
 parser.add_argument(