tangchangli commited on
Commit
298d0a5
1 Parent(s): 0ef6a85

chore: expand max len of model

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -123,7 +123,7 @@ class SALMONN(nn.Module):
123
  prompt,
124
  prompt_pattern="USER: <Speech><SpeechHere></Speech> {}\nASSISTANT:",
125
  device='cuda:0',
126
- max_length=200,
127
  num_beams=4,
128
  do_sample=True,
129
  min_length=1,
 
123
  prompt,
124
  prompt_pattern="USER: <Speech><SpeechHere></Speech> {}\nASSISTANT:",
125
  device='cuda:0',
126
+ max_length=1200,
127
  num_beams=4,
128
  do_sample=True,
129
  min_length=1,