From 10c8ed1d7ecf13bedfa6be5df4ced3665a7eb347 Mon Sep 17 00:00:00 2001
From: IanNod <45800100+IanNod@users.noreply.github.com>
Date: Wed, 13 Dec 2023 14:15:48 -0800
Subject: [PATCH] Small fix to enable gen_external_params.py script to not
 quantize (#247)

---
 .../turbine_models/gen_external_params/gen_external_params.py   | 2 +-
 1 file changed, 1 insertion(+), 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
index df79adcc6..cff729208 100644
--- 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(