(parser)
| 21 | |
| 22 | |
| 23 | def add_quantization_args(parser): |
| 24 | group = parser.add_argument_group("Quantization") |
| 25 | |
| 26 | group.add_argument("--quantization-bit-width", type=int, default=None) |
| 27 | group.add_argument("--from-quantized-checkpoint", action="store_true", help="Loading from a quantized checkpoint") |
| 28 | |
| 29 | |
| 30 | def add_initialization_args(parser): |