Update app.py
Browse files
app.py
CHANGED
|
@@ -18,13 +18,13 @@ pipe = pipeline(
|
|
| 18 |
chunk_length_s=10,
|
| 19 |
device=device,
|
| 20 |
generate_kwargs={
|
| 21 |
-
"no_repeat_ngram_size":
|
| 22 |
"repetition_penalty": 1.15,
|
| 23 |
-
"temperature": 0.
|
| 24 |
-
"top_p": 0.97,
|
| 25 |
-
"top_k":
|
| 26 |
-
"max_new_tokens":
|
| 27 |
-
"do_sample":
|
| 28 |
}
|
| 29 |
)
|
| 30 |
pipe.model.config.forced_decoder_ids = pipe.tokenizer.get_decoder_prompt_ids(language=language, task="transcribe")
|
|
|
|
| 18 |
chunk_length_s=10,
|
| 19 |
device=device,
|
| 20 |
generate_kwargs={
|
| 21 |
+
"no_repeat_ngram_size": 3,
|
| 22 |
"repetition_penalty": 1.15,
|
| 23 |
+
"temperature": 0.3,
|
| 24 |
+
# "top_p": 0.97,
|
| 25 |
+
"top_k": 20,
|
| 26 |
+
"max_new_tokens": 200,
|
| 27 |
+
"do_sample": False
|
| 28 |
}
|
| 29 |
)
|
| 30 |
pipe.model.config.forced_decoder_ids = pipe.tokenizer.get_decoder_prompt_ids(language=language, task="transcribe")
|