From 50eef58c6205913fea5d448cb13ad6ab7b5f551a Mon Sep 17 00:00:00 2001 From: artbataev Date: Wed, 23 Oct 2024 11:13:50 +0000 Subject: [PATCH] Apply isort and black reformatting Signed-off-by: artbataev --- nemo/collections/llm/api.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nemo/collections/llm/api.py b/nemo/collections/llm/api.py index ee0584ba5720..f72b923673bf 100644 --- a/nemo/collections/llm/api.py +++ b/nemo/collections/llm/api.py @@ -293,8 +293,8 @@ def ptq( 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