pszemraj commited on
Commit
b327b2e
β€’
1 Parent(s): 0508364

πŸ› +format

Browse files

Signed-off-by: peter szemraj <peterszemraj@gmail.com>

Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import argparse
 
2
  import logging
3
  import time
4
  import gradio as gr
@@ -42,7 +43,9 @@ def generate_text(
42
  if verbose:
43
  logging.info(f"Generating text from prompt:\n\n{prompt}")
44
  logging.info(
45
- f"params:\tmax_length={gen_length}, num_beams={num_beams}, no_repeat_ngram_size={no_repeat_ngram_size}, length_penalty={length_penalty}, repetition_penalty={repetition_penalty}, abs_max_length={abs_max_length}"
 
 
46
  )
47
  st = time.perf_counter()
48
 
 
1
  import argparse
2
+ import pprint as pp
3
  import logging
4
  import time
5
  import gradio as gr
 
43
  if verbose:
44
  logging.info(f"Generating text from prompt:\n\n{prompt}")
45
  logging.info(
46
+ pp.pformat(
47
+ f"params:\tmax_length={gen_length}, penalty_alpha={penalty_alpha}, top_k={top_k}, no_repeat_ngram_size={no_repeat_ngram_size}, length_penalty={length_penalty}"
48
+ )
49
  )
50
  st = time.perf_counter()
51