Spaces:
Runtime error
Runtime error
Peter
commited on
Commit
·
334ede4
1
Parent(s):
0ef2a3f
👽️ update api
Browse filesSigned-off-by: Peter <[email protected]>
- converse.py +0 -1
- 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
|
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 |
|