diff --git a/examples/quantization/brevitas/quantize_llm.py b/examples/quantization/brevitas/quantize_llm.py index e8514d3b..3477da3a 100644 --- a/examples/quantization/brevitas/quantize_llm.py +++ b/examples/quantization/brevitas/quantize_llm.py @@ -130,9 +130,8 @@ def main(args): parser.add_argument( "--device", type=str, - choices=["cpu", "cuda:0", "auto"], default="auto", - help='Device to run the example on (e.q., "cpu", "cuda:0", "auto"). "auto" will automatically select the device using HuggingFace Accelerate (choices: [%(choices)s], default: %(default)s).', + help='Device to run the example on (e.g., "cpu", "cuda:0", "auto"). "auto" will automatically select the device using HuggingFace Accelerate (default: %(default)s).', ) parser.add_argument( "--onnx-output-path",