Add argument backend to parser.
(parser)
| 386 | |
| 387 | @staticmethod |
| 388 | def backend(parser): |
| 389 | """Add argument backend to parser.""" |
| 390 | |
| 391 | return parser.add_argument('--backend', |
| 392 | type=str, |
| 393 | default='turbomind', |
| 394 | choices=['pytorch', 'turbomind'], |
| 395 | help='Set the inference backend') |
| 396 | |
| 397 | @staticmethod |
| 398 | def stream_output(parser): |
no outgoing calls
no test coverage detected