@@ -263,55 +263,55 @@ def _configure_artifact_inventory_args(parser, verb: str) -> None:
263263
264264# Add CLI Args specific to user prompted generation
265265def _add_generation_args (parser ) -> None :
266- generater_parser = parser .add_argument_group ("Generation Args" , "Configs for generating output based on provided prompt" )
267- generater_parser .add_argument (
266+ generator_parser = parser .add_argument_group ("Generation Args" , "Configs for generating output based on provided prompt" )
267+ generator_parser .add_argument (
268268 "--prompt" ,
269269 type = str ,
270270 default = "Hello, my name is" ,
271271 help = "Input prompt for manual output generation" ,
272272 )
273- generater_parser .add_argument (
273+ generator_parser .add_argument (
274274 "--chat" ,
275275 action = "store_true" ,
276276 help = "Whether to start an interactive chat session" ,
277277 )
278- generater_parser .add_argument (
278+ generator_parser .add_argument (
279279 "--gui" ,
280280 action = "store_true" ,
281281 help = "Whether to use a web UI for an interactive chat session" ,
282282 )
283- generater_parser .add_argument (
283+ generator_parser .add_argument (
284284 "--num-samples" ,
285285 type = int ,
286286 default = 1 ,
287287 help = "Number of samples" ,
288288 )
289- generater_parser .add_argument (
289+ generator_parser .add_argument (
290290 "--max-new-tokens" ,
291291 type = int ,
292292 default = 200 ,
293293 help = "Maximum number of new tokens" ,
294294 )
295- generater_parser .add_argument (
295+ generator_parser .add_argument (
296296 "--top-k" ,
297297 type = int ,
298298 default = 200 ,
299299 help = "Top-k for sampling" ,
300300 )
301- generater_parser .add_argument (
301+ generator_parser .add_argument (
302302 "--temperature" , type = float , default = 0.8 , help = "Temperature for sampling"
303303 )
304- generater_parser .add_argument (
304+ generator_parser .add_argument (
305305 "--compile-prefill" ,
306306 action = "store_true" ,
307307 help = "Whether to compile the prefill. Improves prefill perf, but has higher compile times." ,
308308 )
309- generater_parser .add_argument (
309+ generator_parser .add_argument (
310310 "--sequential-prefill" ,
311311 action = "store_true" ,
312312 help = "Whether to perform prefill sequentially. Only used for model debug." ,
313313 )
314- generater_parser .add_argument (
314+ generator_parser .add_argument (
315315 "--speculate-k" ,
316316 type = int ,
317317 default = 5 ,
0 commit comments