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