From c9c5122d0386f60ed7a01062610e7201864ca75c Mon Sep 17 00:00:00 2001 From: Deep Sanghani Date: Mon, 8 Dec 2025 16:42:10 +0530 Subject: [PATCH] reducto v2 changes --- lexsiai/core/text.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/lexsiai/core/text.py b/lexsiai/core/text.py index 7831a1d..1fa80e7 100644 --- a/lexsiai/core/text.py +++ b/lexsiai/core/text.py @@ -449,8 +449,11 @@ def quantize_model( qbit: int, instance_type: str, tag: Optional[str] = None, + max_seq_len: Optional[int] = 400, + target_layers: Optional[List[str]] = ["Linear"], + ignore_layers: Optional[List[str]] = ["lm_head"], input_column: Optional[str] = None, - no_of_samples: Optional[str] = None, + no_of_samples: Optional[str] = None ): """Quantize Model @@ -473,6 +476,9 @@ def quantize_model( "instance_type": instance_type, "tag": tag, "input_column": input_column, + "max_seq_len": max_seq_len, + "target_layers": target_layers, + "ignore_layers": ignore_layers, "no_of_samples": no_of_samples }