Apply isort and black reformatting

Signed-off-by: janekl <[email protected]>
NVIDIA · Oct 23, 2024 · 7964a8d · 7964a8d
1 parent a956291
commit 7964a8d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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