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 |
|