From c25a3de12ad97315ebc78616ccf090674de6f114 Mon Sep 17 00:00:00 2001 From: Nick Fraser Date: Tue, 5 Nov 2024 13:44:53 +0000 Subject: [PATCH 1/2] Fix (example/brevitas): Allow setting more device options (e.g., 'cuda:1') --- examples/quantization/brevitas/quantize_llm.py | 1 - 1 file changed, 1 deletion(-) diff --git a/examples/quantization/brevitas/quantize_llm.py b/examples/quantization/brevitas/quantize_llm.py index e8514d3b..11a8d3bc 100644 --- a/examples/quantization/brevitas/quantize_llm.py +++ b/examples/quantization/brevitas/quantize_llm.py @@ -130,7 +130,6 @@ 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).', ) From 2a28605b74b079dfc4638157341f373126c3673c Mon Sep 17 00:00:00 2001 From: Nick Fraser Date: Tue, 5 Nov 2024 13:51:29 +0000 Subject: [PATCH 2/2] Fix (example/brevitas): typo --- examples/quantization/brevitas/quantize_llm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/quantization/brevitas/quantize_llm.py b/examples/quantization/brevitas/quantize_llm.py index 11a8d3bc..3477da3a 100644 --- a/examples/quantization/brevitas/quantize_llm.py +++ b/examples/quantization/brevitas/quantize_llm.py @@ -131,7 +131,7 @@ def main(args): "--device", type=str, 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",