Skip to content

Commit b6c5c57

Browse files
committed
remove default value from argument
1 parent e1c952c commit b6c5c57

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

modules/shared.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ def str2bool(v):
6969
parser.add_argument('--cpu', action='store_true', help='Use the CPU to generate text.')
7070
parser.add_argument('--load-in-8bit', action='store_true', help='Load the model with 8-bit precision.')
7171
parser.add_argument('--gptq-bits', type=int, default=0, help='Load a pre-quantized model with specified precision. 2, 3, 4 and 8bit are supported. Currently only works with LLaMA and OPT.')
72-
parser.add_argument('--gptq-model-type', type=str, default='llama', help='Model type of pre-quantized model. Currently only LLaMa and OPT are supported.')
72+
parser.add_argument('--gptq-model-type', type=str, help='Model type of pre-quantized model. Currently only LLaMa and OPT are supported.')
7373
parser.add_argument('--bf16', action='store_true', help='Load the model with bfloat16 precision. Requires NVIDIA Ampere GPU.')
7474
parser.add_argument('--auto-devices', action='store_true', help='Automatically split the model across the available GPU(s) and CPU.')
7575
parser.add_argument('--disk', action='store_true', help='If the model is too large for your GPU(s) and CPU combined, send the remaining layers to the disk.')

0 commit comments

Comments
 (0)