Spaces:

nics-efc
/

C2C_demo

Running

fuvty commited on Nov 5

Commit

17888ba

1 Parent(s): 4d180b8

minor

Files changed (1) hide show

app.py CHANGED Viewed

@@ -372,7 +372,7 @@ class ModelManager:
             'attention_mask': inputs.attention_mask,
             'position_ids': position_ids,
             'streamer': streamer,
-            **self._get_generation_kwargs(max_new_tokens=2048)
         }
         # Run generation in separate thread

             'attention_mask': inputs.attention_mask,
             'position_ids': position_ids,
             'streamer': streamer,
+            **self._get_generation_kwargs(max_new_tokens=self.t2t_answer_max_tokens)
         }
         # Run generation in separate thread