Peter commited on
Commit
334ede4
1 Parent(s): 0ef2a3f

👽️ update api

Browse files

Signed-off-by: Peter <74869040+pszemraj@users.noreply.github.com>

Files changed (2) hide show
  1. converse.py +0 -1
  2. grammar_improve.py +1 -1
converse.py CHANGED
@@ -162,7 +162,6 @@ def gen_response(
162
  repetition_penalty=3.5,
163
  length_penalty=length_penalty,
164
  clean_up_tokenization_spaces=True,
165
- skip_special_tokens=True,
166
  remove_invalid_values=True,
167
  **kwargs,
168
  ) # the likely better beam-less method
 
162
  repetition_penalty=3.5,
163
  length_penalty=length_penalty,
164
  clean_up_tokenization_spaces=True,
 
165
  remove_invalid_values=True,
166
  **kwargs,
167
  ) # the likely better beam-less method
grammar_improve.py CHANGED
@@ -501,7 +501,7 @@ def correct_grammar(
501
  temperature=temp,
502
  )
503
 
504
- tgt_text = tokenizer.batch_decode(translated, skip_special_tokens=True)
505
  rt_min = (time.perf_counter() - st) / 60
506
  print(f"\n\ncorrected in {rt_min} minutes")
507
 
 
501
  temperature=temp,
502
  )
503
 
504
+ tgt_text = tokenizer.batch_decode(translated)
505
  rt_min = (time.perf_counter() - st) / 60
506
  print(f"\n\ncorrected in {rt_min} minutes")
507