Typo in generate_openelm.py

#8
Files changed (1) hide show
  1. generate_openelm.py +2 -2
generate_openelm.py CHANGED
@@ -217,7 +217,7 @@ if __name__ == '__main__':
217
  args = openelm_generate_parser()
218
  prompt = args.prompt
219
 
220
- output_text, genertaion_time = generate(
221
  prompt=prompt,
222
  model=args.model,
223
  device=args.device,
@@ -234,7 +234,7 @@ if __name__ == '__main__':
234
  f'{output_text}\r\n'
235
  f'{"-" * os.get_terminal_size().columns}\r\n'
236
  '\r\nGeneration took'
237
- f'\033[1m\033[92m {round(genertaion_time, 2)} \033[0m'
238
  'seconds.\r\n'
239
  )
240
  print(print_txt)
 
217
  args = openelm_generate_parser()
218
  prompt = args.prompt
219
 
220
+ output_text, generation_time = generate(
221
  prompt=prompt,
222
  model=args.model,
223
  device=args.device,
 
234
  f'{output_text}\r\n'
235
  f'{"-" * os.get_terminal_size().columns}\r\n'
236
  '\r\nGeneration took'
237
+ f'\033[1m\033[92m {round(generation_time, 2)} \033[0m'
238
  'seconds.\r\n'
239
  )
240
  print(print_txt)