tangchangli commited on
Commit
57e646f
1 Parent(s): 5b7b6db

chore: decrease max len

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=150,
127
  num_beams=4,
128
  do_sample=True,
129
  min_length=1,