xujinheng666 commited on
Commit
7ea48f6
·
verified ·
1 Parent(s): bfe1948

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
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": 4,
22
  "repetition_penalty": 1.15,
23
- "temperature": 0.5,
24
- "top_p": 0.97,
25
- "top_k": 40,
26
- "max_new_tokens": 300,
27
- "do_sample": True
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")