pere commited on
Commit
7813820
·
1 Parent(s): 569a668

update test

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -31,6 +31,7 @@ def pipe(file, return_timestamps=False):
31
  chunk_length_s=30,
32
  device=device,
33
  token=auth_token,
 
34
  torch_dtype=torch.float16,
35
  model_kwargs={"attn_implementation": "flash_attention_2"} if FLASH_ATTENTION else {"attn_implementation": "sdpa"},
36
  )
 
31
  chunk_length_s=30,
32
  device=device,
33
  token=auth_token,
34
+ num_beams=5,
35
  torch_dtype=torch.float16,
36
  model_kwargs={"attn_implementation": "flash_attention_2"} if FLASH_ATTENTION else {"attn_implementation": "sdpa"},
37
  )