Spaces:

tsinghua-ee
/

SALMONN-7B-gradio

Sleeping

tangchangli commited on Nov 10, 2023

Commit

57e646f

•

1 Parent(s): 5b7b6db

chore: decrease max len

Files changed (1) hide show

model.py CHANGED Viewed

@@ -123,7 +123,7 @@ class SALMONN(nn.Module):
         prompt,
         prompt_pattern="USER: <Speech><SpeechHere></Speech> {}\nASSISTANT:",
         device='cuda:0',
-        max_length=200,
         num_beams=4,
         do_sample=True,
         min_length=1,

         prompt,
         prompt_pattern="USER: <Speech><SpeechHere></Speech> {}\nASSISTANT:",
         device='cuda:0',
+        max_length=150,
         num_beams=4,
         do_sample=True,
         min_length=1,