From 7964a8d25703e15173a3a4573c72af6123567268 Mon Sep 17 00:00:00 2001 From: janekl Date: Wed, 23 Oct 2024 11:13:04 +0000 Subject: [PATCH] Apply isort and black reformatting Signed-off-by: janekl --- nemo/collections/llm/api.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/nemo/collections/llm/api.py b/nemo/collections/llm/api.py index 62d7a8e0fdeb8..f72b923673bf4 100644 --- a/nemo/collections/llm/api.py +++ b/nemo/collections/llm/api.py @@ -286,15 +286,15 @@ def default_export( @run.cli.entrypoint(name="ptq", namespace="llm") def ptq( - nemo_checkpoint: str, + nemo_checkpoint: str, # TODO: Maybe also create calibration_config for parallel and data settings? calib_tp: int = 1, calib_pp: int = 1, dataset_size: int = 512, batch_size: int = 64, seq_len: int = 128, - quantization_config: QuantizationConfig = default_quantization(), - export_config: ExportConfig = default_export(None), + quantization_config: QuantizationConfig = default_quantization(), + export_config: ExportConfig = default_export(None), ): """ Applies Post-Training Quantization (PTQ) for a model using the specified quantization and export configs. It runs