Update optimum/gptq/quantizer.py

Co-authored-by: fxmarty <9808326+fxmarty@users.noreply.github.com>
huggingface · Dec 12, 2023 · 2c6a6f0 · 2c6a6f0
1 parent c9f105b
commit 2c6a6f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optimum/gptq/quantizer.py b/optimum/gptq/quantizer.py
@@ -84,7 +84,7 @@ def __init__(
         Args:
             bits (`int`):
                 The number of bits to quantize to, supported numbers are (2, 3, 4, 8).
-            dataset (`Union[List[str],str,Any]`, defaults to None):
+            dataset (`Union[List[str], str, Any]`, defaults to `None`):
                 The dataset used for quantization. You can provide your own dataset in a list of string or in a list of tokenized data or
                 just use the original datasets used in GPTQ paper ['wikitext2','c4','c4-new','ptb','ptb-new'].
             group_size (int, defaults to 128):