Spaces:
Runtime error
Runtime error
tangchangli
commited on
Commit
•
57e646f
1
Parent(s):
5b7b6db
chore: decrease max len
Browse files
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=
|
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,
|