Skip to content

Commit

Permalink
Apply isort and black reformatting
Browse files Browse the repository at this point in the history
Signed-off-by: artbataev <[email protected]>
  • Loading branch information
artbataev committed Oct 23, 2024
1 parent b2abcbe commit 50eef58
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions nemo/collections/llm/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 50eef58

Please sign in to comment.